From 205a660d58683cc860c4ef7fb338de1d7fe0d9ec Mon Sep 17 00:00:00 2001 From: Hiroshiba Date: Thu, 29 Aug 2019 00:04:16 +0900 Subject: [PATCH] update config to use GPU --- recipe/config_sr.json | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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 +}