def forward(self, x):
x = self.bn(x)
x = F.elu(x, inplace=True)
x = self.conv(x)
if self.dropout is not None:
x = self.drop(x)
return x
微信
公众号
扫码关注公众号