diff --git a/recipe/config_sr.json b/recipe/config_sr.json index 7d5a2b4..84c9767 100644 --- 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 } -} \ No newline at end of file +}