diff --git a/ppcls/loss/__init__.py b/ppcls/loss/__init__.py index 47dfc246e604791f6bcb841e9a970a739f6f3a62..006b54eb3f7498b3dd860bc143b51488c7aee340 100644 --- a/ppcls/loss/__init__.py +++ b/ppcls/loss/__init__.py @@ -94,7 +94,7 @@ class CombinedLoss(nn.Layer): loss_dict["loss"] = paddle.add_n(list(loss_dict.values())) if self.scaler: - self.scaler.scale(loss_dict["loss"]) + self.scaler(loss_dict["loss"]) return loss_dict