摘要: class Mish(nn.Module): @staticmethod def forward(x): return x * F.softplus(x).tanh() class MemoryEfficientMish(nn.Module): class F(torch.autograd.Func 阅读全文
posted @ 2020-11-19 09:09 zonechen 阅读(741) 评论(0) 推荐(0) 编辑