[NPU] fix fp16 (#5417)

14e72204 · furnace · GitHub · e5e4a620 · 14e72204
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

ppdet/engine/trainer.py ppdet/engine/trainer.py +1 -1

未找到文件。
--- a/ppdet/engine/trainer.py
+++ b/ppdet/engine/trainer.py
@@ -373,7 +373,7 @@ class Trainer(object):
        # enabel auto mixed precision mode
        if self.cfg.get('amp', False):
            scaler = amp.GradScaler(
-                enable=self.cfg.use_gpu, init_loss_scaling=1024)
+                enable=self.cfg.use_gpu or self.cfg.use_npu, init_loss_scaling=1024)

        self.status.update({
            'epoch_id': self.start_epoch,