From a458a62de42552f53925c282eef633d001dc643a Mon Sep 17 00:00:00 2001
From: Zitian Gao <zitian.gao@outlook.com>
Date: Sun, 1 Jun 2025 20:22:00 +0800
Subject: update wandb

---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'train.py')

diff --git a/train.py b/train.py
index 438c582..de0bf90 100644
--- a/train.py
+++ b/train.py
@@ -12,7 +12,7 @@ import pandas as pd
 import numpy as np
 from torch.utils.data import Dataset, DataLoader
 
-# import wandb
+import wandb
 
 from accelerate import Accelerator, DeepSpeedPlugin
 from accelerate.utils import set_seed
-- 
cgit v1.2.3