浏览代码

Remove float64 numpy

/develop/nopreviousactions
Ervin Teng 5 年前
当前提交
7a401feb
共有 1 个文件被更改,包括 2 次插入1 次删除
  1. 3
      ml-agents/mlagents/trainers/optimizer.py

3
ml-agents/mlagents/trainers/optimizer.py


def _make_zero_mem(self, m_size: int, length: int) -> List[np.ndarray]:
return [
np.zeros((m_size)) for i in range(0, length, self.policy.sequence_length)
np.zeros((m_size), dtype=np.float32)
for i in range(0, length, self.policy.sequence_length)
]
正在加载...
取消
保存