We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent cf357f3 commit 46ea81fCopy full SHA for 46ea81f
batchglm/pkg_constants.py
@@ -22,7 +22,7 @@
22
TF_CONFIG_PROTO.gpu_options.allow_growth = True
23
TF_CONFIG_PROTO.graph_options.optimizer_options.global_jit_level = tf.OptimizerOptions.ON_1
24
25
-TF_CONFIG_PROTO.inter_op_parallelism_threads = 0 if TF_NUM_THREADS == 0 else 1
+TF_CONFIG_PROTO.inter_op_parallelism_threads = TF_NUM_THREADS
26
TF_CONFIG_PROTO.intra_op_parallelism_threads = TF_NUM_THREADS
27
28
if TF_NUM_THREADS == 0:
0 commit comments