diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2019-12-19 17:10:09 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2019-12-19 17:10:09 +0100 |
| commit | 5f1b6c981da4f61e7c08024d9f87c5f6f8163403 (patch) | |
| tree | 8f6ac2587dfc5a93cd383806808936d4deb625d0 | |
| parent | e724a1362a32a517999cfd99a1df0a1054695852 (diff) | |
bin
| -rw-r--r-- | app/relay/interpreters.js | 7 | ||||
| -rw-r--r-- | app/relay/modules/biggan.js | 2 |
2 files changed, 7 insertions, 2 deletions
diff --git a/app/relay/interpreters.js b/app/relay/interpreters.js index d892acf..786a5d1 100644 --- a/app/relay/interpreters.js +++ b/app/relay/interpreters.js @@ -23,6 +23,11 @@ export default { params: ['-u'], gpu: true, }, + pytorch_biggan: { + cmd: process.env.PYTORCH_BIGGAN_BIN, + params: ['-u'], + gpu: true, + }, pytorch: { cmd: process.env.PYTORCH_BIN, params: ['-u'], @@ -43,4 +48,4 @@ export default { params: ['-u'], gpu: false, }, -}
\ No newline at end of file +} diff --git a/app/relay/modules/biggan.js b/app/relay/modules/biggan.js index 53bb12e..97042be 100644 --- a/app/relay/modules/biggan.js +++ b/app/relay/modules/biggan.js @@ -4,7 +4,7 @@ import fs from 'fs' const name = 'biggan' const cwd = process.env.BIGGAN_CWD || path.join(process.env.HOME, 'code/' + name + '/') const live = { - type: 'pytorch', + type: 'pytorch_biggan', script: 'inversion/live.py', params: (task) => { console.log(task) |