diff --git a/src/models/transfer_trainer.py b/src/models/transfer_trainer.py index a5e149d..ceb14bd 100644 --- a/src/models/transfer_trainer.py +++ b/src/models/transfer_trainer.py @@ -246,8 +246,7 @@ def setup_optimizers(self) -> Tuple[optim.Optimizer, optim.lr_scheduler._LRSched optimizer, mode='min', factor=0.5, - patience=5, - verbose=True + patience=5 ) return optimizer, scheduler diff --git a/src/training/trainer.py b/src/training/trainer.py index cdb235e..211ff27 100644 --- a/src/training/trainer.py +++ b/src/training/trainer.py @@ -70,8 +70,7 @@ def __init__(self, model: RegimeDetector, training_config: Optional[TrainingConf mode='min', factor=self.config.lr_scheduler_factor, patience=self.config.lr_scheduler_patience, - min_lr=self.config.lr_scheduler_min_lr, - verbose=True + min_lr=self.config.lr_scheduler_min_lr ) # Mixed precision training