From b8ff9e8fa47e354ab9d0e8e514256fdd5c1f9d4b Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Tue, 26 Jun 2018 14:02:37 +0200
Subject: lenny kogginz

---
 app/relay/interpreters.js | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

(limited to 'app/relay/interpreters.js')

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
-- 
cgit v1.2.3-70-g09d2