diff options
| author | Zitian Gao <zitian.gao@outlook.com> | 2025-06-01 20:22:00 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2025-06-01 20:22:00 +0800 |
| commit | a458a62de42552f53925c282eef633d001dc643a (patch) | |
| tree | 7a11cb45df5bf149f47feeb8b482355f7468ede9 /train.py | |
| parent | f368e1296a535cb7505e6e4a1406677d926300b7 (diff) | |
update wandb
Diffstat (limited to 'train.py')
| -rw-r--r-- | train.py | 2 |
1 files changed, 1 insertions, 1 deletions
@@ -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 |