change default TF_NUM_THREADS to '0', i.e. use all available CPU resources

Author: Hoeze

batchglm/pkg_constants.py

@@ -1,8 +1,9 @@
 import os
+import multiprocessing
 
 import tensorflow as tf
 
-TF_NUM_THREADS = int(os.environ.get('TF_NUM_THREADS', 1))
+TF_NUM_THREADS = int(os.environ.get('TF_NUM_THREADS', 0))
 TF_LOOP_PARALLEL_ITERATIONS = int(os.environ.get('TF_LOOP_PARALLEL_ITERATIONS', 10))
 
 XARRAY_NETCDF_ENGINE = "h5netcdf"
@@ -14,3 +15,6 @@
 
 TF_CONFIG_PROTO.inter_op_parallelism_threads = 0 if TF_NUM_THREADS == 0 else 1
 TF_CONFIG_PROTO.intra_op_parallelism_threads = TF_NUM_THREADS
+
+if TF_NUM_THREADS == 0:
+    TF_NUM_THREADS = multiprocessing.cpu_count()