mirrors/PaddleClas
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleClas.git synced 2025-06-03 21:55:06 +08:00
Code Issues Projects Releases Wiki Activity

fix label smooth for paddle1.7

Browse Source
This commit is contained in:
littletomatodonkey 2020-06-09 03:48:23 +00:00
parent 38ad51ca0d
commit c38fd9f9bb
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
ppcls/modeling/loss.py
View File

@ -41,6 +41,7 @@ class Loss(object):
label=one_hot_target, epsilon=self._epsilon, dtype="float32") label=one_hot_target, epsilon=self._epsilon, dtype="float32")
soft_target = fluid.layers.reshape( soft_target = fluid.layers.reshape(
soft_target, shape=[-1, self._class_dim]) soft_target, shape=[-1, self._class_dim])
soft_target.stop_gradient = True
return soft_target return soft_target
def _crossentropy(self, input, target): def _crossentropy(self, input, target):