1 files changed, 1 insertions, 1 deletions
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