fangjiasheng 2 tahun lalu
induk
melakukan
5568c15f48
1 mengubah file dengan 0 tambahan dan 1 penghapusan
  1. 0 1
      BiddingKG/dl/table_head/predict.py

+ 0 - 1
BiddingKG/dl/table_head/predict.py

@@ -27,7 +27,6 @@ else:
 keras_model_path = os.path.abspath(os.path.dirname(__file__)) + "/best.hdf5"
 # keras模型加载预测都使用同一个session、同一个graph,即可多进程推理
 session_conf = tf.ConfigProto(
-    device_count={"CPU": 5},
     intra_op_parallelism_threads=5,
     inter_op_parallelism_threads=5)
 sess = tf.Session(graph=tf.Graph(), config=session_conf)