diff options
| -rw-r--r-- | recipe/config_sr.json | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/recipe/config_sr.json b/recipe/config_sr.json index 3f290d9..e0b62e7 100644 --- a/recipe/config_sr.json +++ b/recipe/config_sr.json @@ -24,7 +24,7 @@ }, "train": { "batchsize": 8, - "gpu": -1, + "gpu": 0, "log_iteration": 250, "snapshot_iteration": 5000 } |