diff --git a/src/training/trainer_new.py b/src/training/trainer_new.py index 6304b03..33188e9 100644 --- a/src/training/trainer_new.py +++ b/src/training/trainer_new.py @@ -292,7 +292,7 @@ class Trainer: def _setup_tracker(self) -> None: """W&B + TensorBoard tracker.""" - LOGGER.info("⚙️ Setup tracker...") + LOGGER.info("⚙️ Setup tracker...") assert self.output_dir is not None and self.full_config is not None self.tracker = ExperimentTracker( output_dir=self.output_dir, @@ -306,7 +306,7 @@ class Trainer: def _build_model(self) -> None: """Build (or load) the encoder model based on the active backbone.""" - LOGGER.info("⚙️ Build loss...") + LOGGER.info("⚙️ Build model...") backbone = self.models_common_cfg.backbone if self.pipeline_cfg.resume_from is not None: