提交 cca4d85a 编写于 作者: littletomatodonkey's avatar littletomatodonkey

rm comment

上级 7cf6e9ae
......@@ -233,7 +233,6 @@ class Reader:
batch_size=batch_size,
shuffle=self.shuffle,
drop_last=True)
loader = DataLoader(
dataset,
batch_sampler=batch_sampler,
......
......@@ -49,7 +49,6 @@ class Loss(object):
input = -F.log_softmax(input, axis=-1)
cost = paddle.reduce_sum(target * input, dim=-1)
else:
# softmax_out = F.softmax(input)
cost = F.cross_entropy(input=input, label=target)
avg_cost = paddle.mean(cost)
return avg_cost
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册