From a33ec46add7edf9810b95d703d1a7241aeb8d9ca Mon Sep 17 00:00:00 2001 From: Schobbejak Date: Thu, 16 May 2024 10:14:38 +0200 Subject: [PATCH] Fix scheduler step delay in torch trainer --- epochalyst/pipeline/model/training/torch_trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/epochalyst/pipeline/model/training/torch_trainer.py b/epochalyst/pipeline/model/training/torch_trainer.py index 99e478f..48b5bbb 100644 --- a/epochalyst/pipeline/model/training/torch_trainer.py +++ b/epochalyst/pipeline/model/training/torch_trainer.py @@ -619,7 +619,7 @@ def _train_one_epoch( # Step the scheduler if self.initialized_scheduler is not None: - self.initialized_scheduler.step(epoch=epoch) + self.initialized_scheduler.step(epoch=epoch + 1) # Remove the cuda cache torch.cuda.empty_cache()