update config to use GPU

Hiroshiba · Aug 28, 2019 · 205a660 · 205a660
1 parent 80eba45
commit 205a660
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/recipe/config_sr.json b/recipe/config_sr.json
@@ -24,8 +24,8 @@
   },
   "train": {
     "batchsize": 8,
-    "gpu": -1,
+    "gpu": 0,
     "log_iteration": 250,
     "snapshot_iteration": 5000
   }
-}
+}