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

rm comment

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