Skip to content

Commit e24132c

Browse files
committed
fix: 🐛 fix a bug in BasicTSRunner
1 parent cd0fd20 commit e24132c

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/basicts/runners/basicts_runner.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -405,7 +405,7 @@ def _train_loop(self) -> None:
405405
"""
406406

407407
if self.training_unit == "epoch":
408-
self.num_steps = (self.num_epochs - self.start_epoch) * self.steps_per_epoch
408+
self.num_steps = self.num_epochs * self.steps_per_epoch
409409
step_pbar = None
410410

411411
else: # training_unit == "step"

0 commit comments

Comments
 (0)