def __call__(self, x, t):
features = F.concat(self.features(x), axis=1)
out = F.average_pooling_2d(features, ksize=7)
out = self.classifier(out)
loss = F.softmax_cross_entropy(out, t)
chainer.report({'loss': loss, 'accuracy': F.accuracy(out, t)}, self)
return loss
评论列表
文章目录