浏览代码

Added enforcement of m_size to be divisible by 2

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

7
ml-agents/mlagents/trainers/tf_policy.py


"The memory size for brain {0} is 0 even "
"though the trainer uses recurrent.".format(brain.brain_name)
)
elif self.m_size % 2 != 0:
raise UnityPolicyException(
"The memory size for brain {0} is {1} "
"but it must be divisible by 2.".format(
brain.brain_name, self.m_size
)
)
self._initialize_tensorflow_references()
self.load = load

正在加载...
取消
保存