Update config.json

config.json

@@ -11,5 +11,5 @@
     "pretrained": true,
     "dropout": 0.5,
     "seed": 42,
-    "device": "cuda"  // or "cpu"
+    "device": "cpu"  // or "cuda"
 }