浏览代码

using torch.set_num_threads

/develop/torch-omp-no-thread
vincentpierre 4 年前
当前提交
d137feab
共有 1 个文件被更改,包括 1 次插入0 次删除
  1. 1
      ml-agents/mlagents/trainers/trainer_controller.py

1
ml-agents/mlagents/trainers/trainer_controller.py


self.trainer_threads: List[threading.Thread] = []
self.kill_trainers = False
torch.set_num_threads(1)
np.random.seed(training_seed)
tf.set_random_seed(training_seed)
if torch is not None:

正在加载...
取消
保存