diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2018-07-19 13:25:32 +0200 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2018-07-19 13:25:32 +0200 |
| commit | 7cbc72e970dd82cb3dd8915e766e12644ac8743a (patch) | |
| tree | 04f6eb819e1e616fac433542e706a873450774bc | |
| parent | 7a9a31f96594c8354f9778808a0d8fdcdaf2441d (diff) | |
scriptz
| -rw-r--r-- | recipe/config_sr.json | 2 | ||||
| -rw-r--r-- | train.py | 2 |
2 files changed, 2 insertions, 2 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 } @@ -79,5 +79,5 @@ trainer.extend(ext, trigger=trigger_snapshot) trainer.extend(extensions.LogReport(trigger=trigger_log)) -save_args(arguments, arguments.output) +# save_args(arguments, arguments.output) trainer.run() |