Skip to content

Commit 722f6d1

Browse files
committed
Update help message of train.sh
1 parent ecfc00c commit 722f6d1

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

python/getalp/wsd/predict.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44

55
def main():
66
parser = argparse.ArgumentParser()
7-
parser.add_argument('--data_path', type=str)
7+
parser.add_argument('--data_path', required=True, type=str)
88
parser.add_argument('--weights', nargs="+", type=str)
99
args = parser.parse_args()
1010

python/getalp/wsd/train.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -4,10 +4,10 @@
44

55
def main():
66
parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
7-
parser.add_argument('--data_path', required=True, type=str, help="Path to the prepared data")
8-
parser.add_argument('--model_path', required=True, type=str, help="Path to where to save the model")
9-
parser.add_argument('--batch_size', nargs="?", type=int, default=100, help="Size of batch")
10-
parser.add_argument('--ensemble_count', nargs="?", type=int, default=8, help="Number of models to train")
7+
parser.add_argument('--data_path', required=True, type=str, help=" ")
8+
parser.add_argument('--model_path', required=True, type=str, help=" ")
9+
parser.add_argument('--batch_size', nargs="?", type=int, default=100, help=" ")
10+
parser.add_argument('--ensemble_count', nargs="?", type=int, default=8, help=" ")
1111
parser.add_argument('--epoch_count', nargs="?", type=int, default=100, help=" ")
1212
parser.add_argument('--eval_frequency', nargs="?", type=int, default=4000, help=" ")
1313
parser.add_argument('--update_frequency', nargs="?", type=int, default=1, help=" ")

0 commit comments

Comments
 (0)