def forward(self, x):
out = self.conv1(F.relu(self.bn1(x)))
out = F.avg_pool2d(out, 2)
return out
微信
公众号
扫码关注公众号
微信
公众号