浏览代码

Remove threaded from trainer_controller

/develop/sac-apex
Ervin Teng 5 年前
当前提交
92158d54
共有 2 个文件被更改,包括 0 次插入3 次删除
  1. 1
      ml-agents/mlagents/trainers/tests/test_simple_rl.py
  2. 2
      ml-agents/mlagents/trainers/trainer_controller.py

1
ml-agents/mlagents/trainers/tests/test_simple_rl.py


sampler_manager=SamplerManager(None),
resampling_interval=None,
save_freq=save_freq,
threaded=False,
)
# Begin training

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


training_seed: int,
sampler_manager: SamplerManager,
resampling_interval: Optional[int],
threaded: bool = True,
):
"""
:param model_path: Path to save the model.

self.sampler_manager = sampler_manager
self.resampling_interval = resampling_interval
self.threaded = threaded
self.trainer_threads: List[threading.Thread] = []
self.kill_trainers = False
np.random.seed(training_seed)

正在加载...
取消
保存