浏览代码

Remove prints

/comms-grad
Ervin Teng 4 年前
当前提交
170c269e
共有 1 个文件被更改,包括 0 次插入1 次删除
  1. 1
      ml-agents/mlagents/trainers/torch/networks.py

1
ml-agents/mlagents/trainers/torch/networks.py


) -> Tuple[
AgentAction, ActionLogProbs, torch.Tensor, Dict[str, torch.Tensor], torch.Tensor
]:
print(len(critic_obs))
if self.use_lstm:
# Use only the back half of memories for critic and actor
actor_mem, critic_mem = torch.split(memories, self.memory_size // 2, dim=-1)

正在加载...
取消
保存