浏览代码

Leaky ReLU

/develop/leakyrelu
Ervin Teng 4 年前
当前提交
5d3ad161
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 2
      ml-agents/mlagents/trainers/torch/layers.py

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


class Swish(torch.nn.Module):
def forward(self, data: torch.Tensor) -> torch.Tensor:
return torch.nn.functional.hardswish(data)
return torch.nn.functional.leaky_relu(data)
class Initialization(Enum):

正在加载...
取消
保存