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

update checkpoint dir name

Browse Source
This commit is contained in:
tanmayshishodia
2024-07-16 16:26:30 +00:00
parent 77b004b1b0
commit 461c2cd194
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
peft/finetune.py
+1 -3
View File
@@ -335,9 +335,7 @@ def finetune(
patience = 0 patience = 0
best_eval_loss = 1e7 best_eval_loss = 1e7
checkpoint_dir = ( checkpoint_dir = f"{checkpoint_dir}/run_{datetime.now().strftime('%Y%m%d_%H%M%S')}_{wandb.run.id}"
f"{checkpoint_dir}/run_ignore_{datetime.now().strftime('%Y%m%d_%H%M%S')}"
)
for epoch in range(num_epochs): for epoch in range(num_epochs):
print(f"Epoch: {epoch + 1}") print(f"Epoch: {epoch + 1}")
train_its = train_batches.as_numpy_iterator() train_its = train_batches.as_numpy_iterator()