浏览代码

fix tf bc test

/develop/action-spec-gym
Andrew Cohen 4 年前
当前提交
7fe7f3fe
共有 1 个文件被更改,包括 2 次插入2 次删除
  1. 4
      ml-agents/mlagents/trainers/tf/components/bc/module.py

4
ml-agents/mlagents/trainers/tf/components/bc/module.py


self.policy.sequence_length_ph: self.policy.sequence_length,
}
feed_dict[self.model.action_in_expert] = mini_batch_demo["actions"]
if self.policy.behavior_spec.is_action_discrete():
if self.policy.action_spec.is_action_discrete():
sum(self.policy.behavior_spec.discrete_action_branches),
sum(self.policy.action_spec.discrete_action_branches),
),
dtype=np.float32,
)

正在加载...
取消
保存