浏览代码

Fix BC module

/develop/add-fire/memoryclass
Ervin Teng 4 年前
当前提交
3387f56d
共有 1 个文件被更改,包括 1 次插入3 次删除
  1. 4
      ml-agents/mlagents/trainers/torch/components/bc/module.py

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


memories = []
if self.policy.use_recurrent:
memories = torch.zeros(
1, self.n_sequences, self.policy.actor_critic.half_mem_size * 2
)
memories = torch.zeros(1, self.n_sequences, self.policy.m_size)
if self.policy.use_vis_obs:
vis_obs = []

正在加载...
取消
保存