summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorZitian Gao <zitian.gao@outlook.com>2025-06-08 17:31:41 +0800
committerGitHub <noreply@github.com>2025-06-08 17:31:41 +0800
commitc8e09829c715b2308d60010f556f9141026485bf (patch)
treebb30adc97eec6d94bb44fed195f9d830dbc97bcb
parenta526bdc8d6e857a534240d5f7cb07b6a8a1f859e (diff)
update args
-rw-r--r--train.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/train.py b/train.py
index de0bf90..d9ab348 100644
--- a/train.py
+++ b/train.py
@@ -39,7 +39,7 @@ def parse_args():
parser.add_argument('--run_name', type=str, default=None, help='Experiment run name')
parser.add_argument('--wandb_project', type=str, default='entropy-maximization-ft', help='W&B project name')
parser.add_argument('--wandb_name', type=str, default=None, help='W&B run name')
- parser.add_argument('--seed', type=int, default=42, help='Random seed')
+ parser.add_argument('--seed', type=int, default=15, help='Random seed')
return parser.parse_args()
class FTDataset(Dataset):