diff --git a/ppcls/modeling/loss.py b/ppcls/modeling/loss.py index c19926e76b01e4ba8d98e1d9931af48a8a33096a..3ae04b11db2e177a2d26bf511e43cb6460aab44e 100644 --- a/ppcls/modeling/loss.py +++ b/ppcls/modeling/loss.py @@ -41,6 +41,7 @@ class Loss(object): label=one_hot_target, epsilon=self._epsilon, dtype="float32") soft_target = fluid.layers.reshape( soft_target, shape=[-1, self._class_dim]) + soft_target.stop_gradient = True return soft_target def _crossentropy(self, input, target):