diff options
| -rw-r--r-- | app/relay/modules/pix2pixhd.js | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/app/relay/modules/pix2pixhd.js b/app/relay/modules/pix2pixhd.js index 194d9ef..4d8683d 100644 --- a/app/relay/modules/pix2pixhd.js +++ b/app/relay/modules/pix2pixhd.js @@ -58,7 +58,7 @@ const train = { '--dataroot', datasets_path, '--module_name', task.module, '--name', task.dataset, - '--model', 'pix2pixhd', + '--model', 'pix2pixHD', '--label_nc', 0, '--no_instance', '--niter', task.epochs, '--niter_decay', 0, |