Pikaliov/caption-test
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

trainer_new logger small fix

Browse Source
This commit is contained in:
pikaliov
2026-05-07 13:24:01 +03:00
parent 676dea7932
commit 1c03811af0
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/training/trainer_new.py
View File

@@ -292,7 +292,7 @@ class Trainer:
def _setup_tracker(self) -> None: def _setup_tracker(self) -> None:
"""W&B + TensorBoard tracker.""" """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 assert self.output_dir is not None and self.full_config is not None
self.tracker = ExperimentTracker( self.tracker = ExperimentTracker(
output_dir=self.output_dir, output_dir=self.output_dir,
@@ -306,7 +306,7 @@ class Trainer:
def _build_model(self) -> None: def _build_model(self) -> None:
"""Build (or load) the encoder model based on the active backbone.""" """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 backbone = self.models_common_cfg.backbone
if self.pipeline_cfg.resume_from is not None: if self.pipeline_cfg.resume_from is not None: