def backprop(self,t,x=None):
if x is None:
x=Tensor.context
#loss = F.mean_squared_error(x.content,t.content)
loss = F.softmax_cross_entropy(x.content,t.content)
if Deel.train:
loss.backward()
accuracy = F.accuracy(x.content,t.content)
self.optimizer.update()
return loss.data,accuracy.data
评论列表
文章目录