浏览代码

revert changes

/develop/torch-omp-no-thread
Ruo-Ping Dong 4 年前
当前提交
7eceb27f
共有 2 个文件被更改,包括 2 次插入2 次删除
  1. 2
      ml-agents/mlagents/trainers/learn.py
  2. 2
      ml-agents/mlagents/trainers/settings.py

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


param_manager=env_parameter_manager,
init_path=maybe_init_path,
multi_gpu=False,
force_torch=force_torch,
force_torch="torch" in DetectDefault.non_default_args,
)
# Create controller and begin training.
tc = TrainerController(

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


threaded: bool = True
self_play: Optional[SelfPlaySettings] = None
behavioral_cloning: Optional[BehavioralCloningSettings] = None
framework: FrameworkType = FrameworkType.PYTORCH
framework: FrameworkType = FrameworkType.TENSORFLOW
cattr.register_structure_hook(
Dict[RewardSignalType, RewardSignalSettings], RewardSignalSettings.structure

正在加载...
取消
保存