summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-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):