summaryrefslogtreecommitdiff
path: root/recipe
diff options
context:
space:
mode:
Diffstat (limited to 'recipe')
-rw-r--r--recipe/config_sr.json2
1 files changed, 1 insertions, 1 deletions
diff --git a/recipe/config_sr.json b/recipe/config_sr.json
index e0b62e7..06c5e47 100644
--- a/recipe/config_sr.json
+++ b/recipe/config_sr.json
@@ -24,7 +24,7 @@
},
"train": {
"batchsize": 8,
- "gpu": 0,
+ "gpu": 1,
"log_iteration": 250,
"snapshot_iteration": 5000
}