浏览代码

Fix layers reference

/develop/add-fire/test-simple-rl-fix-resnet
Ervin Teng 4 年前
当前提交
d096f69c
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 2
      ml-agents/mlagents/trainers/torch/encoders.py

2
ml-agents/mlagents/trainers/torch/encoders.py


kernel_init=Initialization.KaimingHeNormal,
kernel_gain=1.0,
)
self.sequential = nn.Sequential(*self.layers)
self.sequential = nn.Sequential(*layers)
def forward(self, visual_obs):
batch_size = visual_obs.shape[0]

正在加载...
取消
保存