From ae395151f5700c9d0c6c22a80dafe4d9c6b4f8be Mon Sep 17 00:00:00 2001 From: mo10107 <67790368+mo10107@users.noreply.github.com> Date: Fri, 25 Jul 2025 10:51:05 +0300 Subject: [PATCH] fix(training): remove verbose param from ReduceLROnPlateau to avoid TypeError --- vecman/utils/training.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vecman/utils/training.py b/vecman/utils/training.py index ce0f85d..af27042 100644 --- a/vecman/utils/training.py +++ b/vecman/utils/training.py @@ -60,7 +60,7 @@ def _train_loop(model: VQVAE, loader: DataLoader, epochs: int, device: str, lear opt = optim.Adam(model.parameters(), lr=learning_rate) scheduler = optim.lr_scheduler.ReduceLROnPlateau( - opt, mode='min', factor=0.8, patience=2, verbose=True + opt, mode='min', factor=0.8, patience=2 ) best_loss = float('inf')