diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2018-06-16 21:23:21 +0200 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2018-06-16 21:23:21 +0200 |
| commit | 0d3580fe118e9187dbc88bb28bd3ed9078d96fa9 (patch) | |
| tree | 7953ab749fcc406f26b9e0e9d2d7d9d34089317e /app/relay/interpreters.js | |
| parent | 5c1441fd70f2967ae978a0a83ffd73ed4ac51ea0 (diff) | |
module
Diffstat (limited to 'app/relay/interpreters.js')
| -rw-r--r-- | app/relay/interpreters.js | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/app/relay/interpreters.js b/app/relay/interpreters.js index 90dfcaa..4671988 100644 --- a/app/relay/interpreters.js +++ b/app/relay/interpreters.js @@ -19,6 +19,11 @@ export default { params: ['-u'], gpu: true, }, + pytorch_cpu: { + cmd: process.env.PYTORCH_BIN, + params: ['-u'], + gpu: false, + }, tensorflow: { cmd: process.env.TENSORFLOW_BIN, params: ['-u'], |