diff options
Diffstat (limited to 'recipe')
| -rw-r--r-- | recipe/config_sr.json | 2 | ||||
| -rw-r--r-- | recipe/recipe.json | 6 |
2 files changed, 4 insertions, 4 deletions
diff --git a/recipe/config_sr.json b/recipe/config_sr.json index 06c5e47..3f290d9 100644 --- a/recipe/config_sr.json +++ b/recipe/config_sr.json @@ -24,7 +24,7 @@ }, "train": { "batchsize": 8, - "gpu": 1, + "gpu": -1, "log_iteration": 250, "snapshot_iteration": 5000 } diff --git a/recipe/recipe.json b/recipe/recipe.json index f0bc5d9..d3b8df1 100644 --- a/recipe/recipe.json +++ b/recipe/recipe.json @@ -20,9 +20,9 @@ ], "train/gpu": [ 0, - 1, - 2, - 3 + 0, + 0, + 0, ], "project/name": [ "pp-weakD-el8", |