Fix Wandb errro

This commit is contained in:
misha-chertushkin
2025-01-21 01:50:18 +00:00
parent 4942cda83b
commit fb6213ed59
2 changed files with 2 additions and 2 deletions
+1 -1
View File
@@ -90,7 +90,7 @@ class TimesFMFinetuner:
def _setup_wandb(self) -> None:
"""Initialize Weights & Biases logging."""
wandb.init(project=self.config.wandb_project, entity=self.config.wandb_entity, config=self.config.__dict__)
wandb.init(project=self.config.wandb_project, config=self.config.__dict__)
def _create_dataloader(self, dataset: Dataset, name: str) -> DataLoader:
"""Create a dataloader from a dataset."""