Fix epoch number

Author: MGlauer

chebai/models/base.py

@@ -211,6 +211,7 @@ def run(
         trainer = pl.Trainer(
             logger=tb_logger,
             min_epochs=epochs,
+            max_epochs=epochs,
             callbacks=[best_checkpoint_callback, checkpoint_callback, es],
             replace_sampler_ddp=False,
             **trainer_kwargs