diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2018-06-26 14:02:37 +0200 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2018-06-26 14:02:37 +0200 |
| commit | b8ff9e8fa47e354ab9d0e8e514256fdd5c1f9d4b (patch) | |
| tree | c8c4a5728100679e3344dde4473e4aed65944058 /app/relay/interpreters.js | |
| parent | 17189b05d4c6f1f0be6541b4970931e39aec17e7 (diff) | |
lenny kogginz
Diffstat (limited to 'app/relay/interpreters.js')
| -rw-r--r-- | app/relay/interpreters.js | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/app/relay/interpreters.js b/app/relay/interpreters.js index 4671988..2cc8622 100644 --- a/app/relay/interpreters.js +++ b/app/relay/interpreters.js @@ -19,7 +19,7 @@ export default { params: ['-u'], gpu: true, }, - pytorch_cpu: { + pytorch_cpu: { // "cpu" pytorch tasks can use the gpu, but not intensively cmd: process.env.PYTORCH_BIN, params: ['-u'], gpu: false, @@ -29,4 +29,9 @@ export default { params: ['-u'], gpu: true, }, + tensorflow_cpu: { + cmd: process.env.TENSORFLOW_BIN, + params: ['-u'], + gpu: false, + }, }
\ No newline at end of file |