浏览代码

fix specs in torch util

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

2
ml-agents/mlagents/trainers/torch/utils.py


@property
def flattened_size(self) -> int:
if self._specs.is_continuous():
return self._specs.size
return self._specs.continuous_size
else:
return sum(self._specs.discrete_branches)

正在加载...
取消
保存