浏览代码

fixing typo

/develop/add-fire
vincentpierre 4 年前
当前提交
eb951ca0
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 2
      ml-agents/mlagents/trainers/torch/encoders.py

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


self.layers.append(Swish())
self.seq_layers = nn.Sequential(*self.layers)
def forward(self, inputs: torch.Tensor) -> None:
def forward(self, inputs: torch.Tensor) -> torch.Tensor:
if self.normalizer is not None:
inputs = self.normalizer(inputs)
return self.seq_layers(inputs)

正在加载...
取消
保存