freedak/timesfm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix Wandb errro

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files
notebooks/finetuning_example.py
+1 -1
View File
@@ -202,7 +202,7 @@ def get_data(context_len: int, horizon_len: int) -> Tuple[Dataset, Dataset]:
def basic_example(): def basic_example():
"""Basic example of finetuning TimesFM on stock data.""" """Basic example of finetuning TimesFM on stock data."""
model, hparams, tfm_config = get_model(load_weights=True) 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) train_dataset, val_dataset = get_data(128, tfm_config.horizon_len)
finetuner = TimesFMFinetuner(model, config) finetuner = TimesFMFinetuner(model, config)
notebooks/finetuning_torch.py
+1 -1
View File
@@ -90,7 +90,7 @@ class TimesFMFinetuner:
def _setup_wandb(self) -> None: def _setup_wandb(self) -> None:
"""Initialize Weights & Biases logging.""" """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: def _create_dataloader(self, dataset: Dataset, name: str) -> DataLoader:
"""Create a dataloader from a dataset.""" """Create a dataloader from a dataset."""