Removed logging from WeightAveraging

Author: Seppo Enarvi

src/lightning/pytorch/callbacks/weight_averaging.py

@@ -190,7 +190,6 @@ def on_train_end(self, trainer: "pl.Trainer", pl_module: "pl.LightningModule") -
 
         """
         assert self._average_model is not None
-        rank_zero_info("Loading the average model parameters to the final model.")
         self._copy_average_to_current(pl_module)
 
     @override
@@ -205,7 +204,6 @@ def on_validation_epoch_start(self, trainer: "pl.Trainer", pl_module: "pl.Lightn
 
         """
         if self._average_model is not None:
-            rank_zero_info("Loading the average model parameters for validation.")
             self._swap_models(pl_module)
 
     @override
@@ -220,7 +218,6 @@ def on_validation_epoch_end(self, trainer: "pl.Trainer", pl_module: "pl.Lightnin
 
         """
         if self._average_model is not None:
-            rank_zero_info("Recovering the current model parameters after validation.")
             self._swap_models(pl_module)
 
     @override
@@ -270,7 +267,6 @@ def on_save_checkpoint(
                 "average model parameters will be saved to the state_dict in the checkpoint."
             )
         else:
-            rank_zero_info("The average model parameters will be saved to the state_dict in the checkpoint.")
             average_model_state = self._average_model.state_dict()
             checkpoint["current_model_state"] = checkpoint["state_dict"]
             checkpoint["state_dict"] = {