summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--app/relay/modules/pix2pix.js3
1 files changed, 2 insertions, 1 deletions
diff --git a/app/relay/modules/pix2pix.js b/app/relay/modules/pix2pix.js
index 0e67621..f258620 100644
--- a/app/relay/modules/pix2pix.js
+++ b/app/relay/modules/pix2pix.js
@@ -93,8 +93,9 @@ const train = {
args = args.concat([
'--which_epoch', 'latest',
'--continue_train',
- })
+ ])
}
+ return args
},
}
const generate = {