use self.core

luxonis · Nov 15, 2024 · 6e0c888 · 6e0c888
1 parent 147c1ee
commit 6e0c888
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/luxonis_train/models/luxonis_lightning.py b/luxonis_train/models/luxonis_lightning.py
@@ -865,7 +865,7 @@ def configure_callbacks(self) -> list[pl.Callback]:
                 )
 
         if self.training_strategy is not None:
-            callbacks.append(TrainingManager(strategy=self.training_strategy))
+            callbacks.append(TrainingManager(strategy=self.training_strategy))  # type: ignore
 
         return callbacks
 

diff --git a/luxonis_train/strategies/triple_lr_sgd.py b/luxonis_train/strategies/triple_lr_sgd.py
@@ -155,8 +155,7 @@ def __init__(self, pl_module: pl.LightningModule, params: dict):
         self.cfg = self.model.cfg
 
         max_stepnum = math.ceil(
-            len(self.model._core.loaders["train"])
-            / self.cfg.trainer.batch_size
+            len(self.model.core.loaders["train"]) / self.cfg.trainer.batch_size
         )
 
         self.optimizer = TripleLRSGD(self.model, params).create_optimizer()