From 363c7467bbdef59feac3ce39c936379283526205 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Sat, 16 Jun 2018 18:03:26 +0200 Subject: epochs --- app/relay/modules/pix2pix.js | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'app/relay/modules/pix2pix.js') 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', ]) -- cgit v1.2.3-70-g09d2