浏览代码

Fix brain name bug in offline bc (#1395)

/develop-generalizationTraining-TrainerController
GitHub 6 年前
当前提交
48578199
共有 1 个文件被更改,包括 5 次插入3 次删除
  1. 8
      ml-agents/mlagents/trainers/bc/offline_trainer.py

8
ml-agents/mlagents/trainers/bc/offline_trainer.py


trainer_parameters['demo_path'],
self.policy.sequence_length)
print(brain.__dict__)
print(brain_params.__dict__)
if brain.__dict__ != brain_params.__dict__:
policy_brain = brain.__dict__
expert_brain = brain_params.__dict__
policy_brain.pop('brain_name')
expert_brain.pop('brain_name')
if expert_brain != policy_brain:
raise UnityTrainerException("The provided demonstration is not compatible with the "
"brain being used for performance evaluation.")

正在加载...
取消
保存