From cb732b1c8fea70e27bf00bfd729f9ad6a8cca446 Mon Sep 17 00:00:00 2001
From: Nikolai <osterei33@gmx.de>
Date: Mon, 13 Aug 2018 09:23:59 +0200
Subject: [PATCH] commenting number of cores for session
---
toolkit.py | 18 +++++++++---------
1 file changed, 9 insertions(+), 9 deletions(-)
diff --git a/toolkit.py b/toolkit.py
index 3ef7399..05534a2 100755
--- a/toolkit.py
+++ b/toolkit.py
@@ -43,15 +43,15 @@ from .utils import WeightedRobustScaler, weighted_quantile
# configure number of cores
# this doesn't seem to work, but at least with these settings keras only uses 4 processes
-import tensorflow as tf
-from keras import backend as K
-num_cores = 1
-config = tf.ConfigProto(intra_op_parallelism_threads=num_cores,
- inter_op_parallelism_threads=num_cores,
- allow_soft_placement=True,
- device_count = {'CPU': num_cores})
-session = tf.Session(config=config)
-K.set_session(session)
+# import tensorflow as tf
+# from keras import backend as K
+# num_cores = 1
+# config = tf.ConfigProto(intra_op_parallelism_threads=num_cores,
+# inter_op_parallelism_threads=num_cores,
+# allow_soft_placement=True,
+# device_count = {'CPU': num_cores})
+# session = tf.Session(config=config)
+# K.set_session(session)
import ROOT
--
GitLab