From 0d3580fe118e9187dbc88bb28bd3ed9078d96fa9 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Sat, 16 Jun 2018 21:23:21 +0200 Subject: module --- app/relay/interpreters.js | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'app/relay/interpreters.js') 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'], -- cgit v1.2.3-70-g09d2