tensorrt path

1 files changed, 5 insertions, 7 deletions

diff --git a/app/relay/modules/pix2pixhd.js b/app/relay/modules/pix2pixhd.js
index b35dbd9..1c7f498 100644
--- a/app/relay/modules/pix2pixhd.js
+++ b/app/relay/modules/pix2pixhd.js
@@ -104,6 +104,7 @@ const live = {
     console.log(task)
     const opt = task.opt || {}
     return [
+      '--phase', 'recursive',
       '--dataroot', path.join(cwd, 'sequences', task.dataset),
       '--start_img', path.join(cwd, 'sequences', task.dataset, 'frame_00001.png'),
       '--checkpoint-name', task.checkpoint,
@@ -114,21 +115,18 @@ const live = {
       '--recursive', '--recursive-frac', 0.1,
       '--sequence', '--sequence-frac', 0.3,
       '--process-frac', 0.5,
-      '--nThreads', 0,
-      '--transition-min', 0.05,
+      '--label_nc', '0', '--no_instance',
       '--how_many', 1000000, '--transition-period', 1000,
-      '--loadSize', 256, '--fineSize', 256,
       '--just-copy', '--poll_delay', opt.poll_delay || 0.09,
-      '--model', 'test',
-      '--which_model_netG', 'unet_256',
-      '--which_direction', 'AtoB',
-      '--dataset_mode', 'recursive',
       '--which_epoch', 'latest',
       '--norm', 'batch',
     ]
   },
 }
 
+
+
+
 export default {
   name, cwd, env,
   activities: {