diff options
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 63a2c25..90dfcaa 100644 --- a/app/relay/interpreters.js +++ b/app/relay/interpreters.js @@ -1,3 +1,5 @@ +require('dotenv').config() + export default { bash: { cmd: process.env.BASH_BIN || '/bin/bash', @@ -8,15 +10,18 @@ export default { gpu: false, }, python: { - cmd: process.env.PYTHON_BIN || '/usr/bin/python3', + cmd: process.env.PYTHON_BIN || '/usr/bin/python', + params: ['-u'], gpu: false, }, pytorch: { cmd: process.env.PYTORCH_BIN, + params: ['-u'], gpu: true, }, tensorflow: { cmd: process.env.TENSORFLOW_BIN, + params: ['-u'], gpu: true, }, }
\ No newline at end of file |