From a446b75333ebf0216a0bfc6e345fdee553a7e1eb Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Mon, 18 Jun 2018 15:57:54 +0200 Subject: img opz --- app/relay/modules/pix2pixhd.js | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) (limited to 'app/relay/modules/pix2pixhd.js') diff --git a/app/relay/modules/pix2pixhd.js b/app/relay/modules/pix2pixhd.js index 5b9fd5c..34fc972 100644 --- a/app/relay/modules/pix2pixhd.js +++ b/app/relay/modules/pix2pixhd.js @@ -3,6 +3,10 @@ import fs from 'fs' const name = 'pix2pixhd' const cwd = process.env.PIX2PIXHD_CWD || path.join(process.env.HOME, 'code/' + name + '/') +const env = { + LD_LIBRARY_PATH: process.env.HOME + '/Downloads/TensorRT-3.0.4/lib', +} +console.log(env) const fetch = { type: 'perl', @@ -101,7 +105,6 @@ const live = { console.log(task) const opt = task.opt || {} return [ - 'LD_LIBRARY_PATH=' + process.env.HOME + '/Downloads/TensorRT-3.0.4/lib', '--dataroot', path.join(cwd, 'sequences', task.dataset), '--start_img', path.join(cwd, 'sequences', task.dataset, 'frame_00001.png'), '--checkpoint-name', task.checkpoint, @@ -128,7 +131,7 @@ const live = { } export default { - name, cwd, + name, cwd, env, activities: { fetch, build, -- cgit v1.2.3-70-g09d2