From fb6213ed59313d5f73b95916aa9e92dadb87ccef Mon Sep 17 00:00:00 2001 From: misha-chertushkin Date: Tue, 21 Jan 2025 01:50:18 +0000 Subject: [PATCH] Fix Wandb errro --- notebooks/finetuning_example.py | 2 +- notebooks/finetuning_torch.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/notebooks/finetuning_example.py b/notebooks/finetuning_example.py index 5664456..966c980 100644 --- a/notebooks/finetuning_example.py +++ b/notebooks/finetuning_example.py @@ -202,7 +202,7 @@ def get_data(context_len: int, horizon_len: int) -> Tuple[Dataset, Dataset]: def basic_example(): """Basic example of finetuning TimesFM on stock data.""" model, hparams, tfm_config = get_model(load_weights=True) - config = FinetuningConfig(batch_size=256, num_epochs=5, learning_rate=1e-4, use_wandb=False) + config = FinetuningConfig(batch_size=256, num_epochs=5, learning_rate=1e-4, use_wandb=True) train_dataset, val_dataset = get_data(128, tfm_config.horizon_len) finetuner = TimesFMFinetuner(model, config) diff --git a/notebooks/finetuning_torch.py b/notebooks/finetuning_torch.py index bf908e7..e923136 100644 --- a/notebooks/finetuning_torch.py +++ b/notebooks/finetuning_torch.py @@ -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."""