浏览代码

all bc tests

/develop/add-fire/bc
Andrew Cohen 4 年前
当前提交
742940a3
共有 1 个文件被更改,包括 4 次插入7 次删除
  1. 11
      ml-agents/mlagents/trainers/torch/components/bc/module.py

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


)
)
memories = [
ModelUtils.list_to_tensor(mini_batch_demo["memory"][i])
for i in range(
0, len(mini_batch_demo["memory"]), self.policy.sequence_length
memories = []
if self.policy.actor_critic.use_lstm:
memories = torch.zeros(
1, self.n_sequences, self.policy.actor_critic.half_mem_size * 2
]
if len(memories) > 0:
memories = torch.stack(memories).unsqueeze(0)
if self.policy.use_vis_obs:
vis_obs = []

正在加载...
取消
保存