diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2020-01-10 03:03:25 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2020-01-10 03:03:25 +0100 |
| commit | 816321e4d4ee481e5cb3bafedd2366ec9fdd6af1 (patch) | |
| tree | 384b006c9cd2cb27fd4b18016d3adbbdb89e1b37 /app/relay/modules/biggan.js | |
| parent | 2c110b07b5fc56d70587b13bec2c08536684ea93 (diff) | |
biggan path
Diffstat (limited to 'app/relay/modules/biggan.js')
| -rw-r--r-- | app/relay/modules/biggan.js | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/app/relay/modules/biggan.js b/app/relay/modules/biggan.js index 6d6e211..44e8e4e 100644 --- a/app/relay/modules/biggan.js +++ b/app/relay/modules/biggan.js @@ -2,7 +2,7 @@ import path from 'path' import fs from 'fs' const name = 'biggan' -const cwd = process.env.BIGGAN_CWD || path.join(process.env.HOME, 'code/' + name + '/') +const cwd = process.env.BIGGAN_CWD || path.join(process.env.HOME, 'code/' + name + '/cli/') const env = { LD_LIBRARY_PATH: '/usr/local/cuda/lib64:' + process.env.TENSORRT_LIB_PATH, TFHUB_CACHE_DIR: '/home/lens/.tfhub', @@ -11,7 +11,7 @@ const env = { } const live = { type: 'pytorch_biggan', - script: 'cli/cli.py', + script: 'cli.py', params: (task) => { console.log(task) const opt = task.opt || {} |
