update config to use GPU

Hiroshiba · Aug 28, 2019 · 59d43db · 59d43db
1 parent 205a660
commit 59d43db
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/recipe/config.json b/recipe/config.json
@@ -37,7 +37,7 @@
   },
   "train": {
     "batchsize": 8,
-    "gpu": -1,
+    "gpu": 0,
     "log_iteration": 250,
     "snapshot_iteration": 5000
   }