def forward(self, input):
out = self.init_cnn_layer(input)
out = self.denseblocks(out)
out = F.avg_pool2d(out, 8).squeeze()
return self.lr(out)
微信
公众号
扫码关注公众号