diff --git a/toolkit.py b/toolkit.py index 3ef7399233d94897d311bf708165927a6149e4d0..05534a2c66836acb048ed47879d0f1d3bafc8ca0 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