AMP log message

AdvancedPhotonSource · Apr 11, 2024 · 62417a3 · 62417a3
1 parent 279d4f5
commit 62417a3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/generic_trainer/trainer.py b/generic_trainer/trainer.py
@@ -742,6 +742,7 @@ def build_amp(self):
                                        HuggingFaceAcceleratePretrainer,
                                        PyTorchLightningTrainer]):
             self.use_torch_amp = True
+            logging.info('Using PyTorch AMP and gradient scaler.')
         self.grad_scaler = torch.cuda.amp.GradScaler(enabled=self.use_torch_amp)
 
     def build_model(self):