diff options
Diffstat (limited to 'app/relay/modules/pix2pix.js')
| -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', ]) |