未验证 提交 0baa9b36 编写于 作者: C ceci3 提交者: GitHub

fix bug (#737)

上级 4f99291e
......@@ -52,7 +52,7 @@ def build_program(archs):
output = static.nn.fc(output, size=10)
softmax_out = F.softmax(output)
cost = F.cross_entropy(softmax_out, label=label)
cost = F.cross_entropy(softmax_out, label=gt)
avg_cost = paddle.mean(cost)
acc_top1 = paddle.metric.accuracy(input=softmax_out, label=gt, k=1)
acc_top5 = paddle.metric.accuracy(input=softmax_out, label=gt, k=5)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册