def forward(self, x):
out = self.pool(self.conv(F.relu(self.bn(x))))
if self.p > 0:
out = F.dropout(out, p=self.p, training=self.training)
return out
微信
公众号
扫码关注公众号