Update train_d2q.py

lingwei-gu · Jan 3, 2022 · 61da174 · 61da174
1 parent 841e232
commit 61da174
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/train_d2q.py b/scripts/train_d2q.py
@@ -32,7 +32,7 @@ def __getitem__(self, idx):
 
 parser = argparse.ArgumentParser(description='Train docTquery on more datasets')
 parser.add_argument('--pretrained_model_path', default='t5-base', help='pretrained model path')
-parser.add_argument('--tag', defaut='msmarco', help='tag for training data', type=str)
+parser.add_argument('--tag', default='msmarco', help='tag for training data', type=str)
 parser.add_argument('--train_data_path', required=True, help='training data path')
 parser.add_argument('--output_path', required=True, help='output directory path')
 parser.add_argument('--epoch', default=1, type=int)