From 461c2cd19470a9f78a8a32d24c9aa5ae534c7bb7 Mon Sep 17 00:00:00 2001
From: tanmayshishodia <tanmay.shishodia@nutanix.com>
Date: Tue, 16 Jul 2024 16:26:30 +0000
Subject: [PATCH] update checkpoint dir name

---
 peft/finetune.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/peft/finetune.py b/peft/finetune.py
index 747dac5..d239530 100644
--- a/peft/finetune.py
+++ b/peft/finetune.py
@@ -335,9 +335,7 @@ def finetune(
 
     patience = 0
     best_eval_loss = 1e7
-    checkpoint_dir = (
-        f"{checkpoint_dir}/run_ignore_{datetime.now().strftime('%Y%m%d_%H%M%S')}"
-    )
+    checkpoint_dir = f"{checkpoint_dir}/run_{datetime.now().strftime('%Y%m%d_%H%M%S')}_{wandb.run.id}"
     for epoch in range(num_epochs):
         print(f"Epoch: {epoch + 1}")
         train_its = train_batches.as_numpy_iterator()