浏览代码

Add warning if behavior name not found in trainer config (#4204)

Co-authored-by: Ervin T. <ervin@unity3d.com>
Co-authored-by: Chris Elion <chris.elion@unity3d.com>
/MLA-1734-demo-provider
GitHub 4 年前
当前提交
f871ea95
共有 1 个文件被更改,包括 5 次插入0 次删除
  1. 5
      ml-agents/mlagents/trainers/trainer_util.py

5
ml-agents/mlagents/trainers/trainer_util.py


self.ghost_controller = GhostController()
def generate(self, brain_name: str) -> Trainer:
if brain_name not in self.trainer_config.keys():
logger.warning(
f"Behavior name {brain_name} does not match any behaviors specified in the trainer configuration file:"
f"{sorted(self.trainer_config.keys())}"
)
return initialize_trainer(
self.trainer_config[brain_name],
brain_name,

正在加载...
取消
保存