diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2018-06-16 18:03:26 +0200 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2018-06-16 18:03:26 +0200 |
| commit | 363c7467bbdef59feac3ce39c936379283526205 (patch) | |
| tree | 731706e074d881a0ea9573bb3ac6759d0f8e3d6e | |
| parent | 3498906ca6df57651118d8af11eb41204cfbeaa9 (diff) | |
epochs
| -rw-r--r-- | app/relay/modules/pix2pix.js | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/app/relay/modules/pix2pix.js b/app/relay/modules/pix2pix.js index f258620..67cca3e 100644 --- a/app/relay/modules/pix2pix.js +++ b/app/relay/modules/pix2pix.js @@ -82,7 +82,6 @@ const train = { '--which_direction', 'AtoB', '--lambda_B', 100, '--dataset_mode', 'aligned', - '--epoch_count', task.epoch + task.epochs + 1, '--niter', task.epochs, '--niter_decay', 0, '--no_lsgan', @@ -91,6 +90,7 @@ const train = { ] if (epoch) { args = args.concat([ + '--epoch_count', task.epoch + task.epochs + 1, '--which_epoch', 'latest', '--continue_train', ]) |