浏览代码

fix policy action_spec typo

/MLA-1734-demo-provider
Ruo-Ping Dong 4 年前
当前提交
d3348074
共有 1 个文件被更改,包括 4 次插入4 次删除
  1. 8
      ml-agents/mlagents/trainers/torch/model_serialization.py

8
ml-agents/mlagents/trainers/torch/model_serialization.py


)
self.output_names = ["version_number", "memory_size"]
if self.policy.action_spec.continuous_size > 0:
if self.policy.behavior_spec.action_spec.continuous_size > 0:
if self.policy.action_spec.discrete_size > 0:
if self.policy.behavior_spec.action_spec.discrete_size > 0:
self.policy.action_spec.continuous_size == 0
or self.policy.action_spec.discrete_size == 0
self.policy.behavior_spec.action_spec.continuous_size == 0
or self.policy.behavior_spec.action_spec.discrete_size == 0
):
self.output_names += [
"action",

正在加载...
取消
保存