In train.yaml remove seq_length and epoch arguments

matsuobasho · Oct 12, 2023 · 6aef523 · 6aef523
1 parent d6ae625
commit 6aef523
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/codegen_model_comparison/src/train.py b/codegen_model_comparison/src/train.py
@@ -102,12 +102,12 @@ def main(args):
 
 def parse_args():
     parser = argparse.ArgumentParser()
-    parser.add_arguments("--data_path")
-    parser.add_arguments("--output_path")
-    parser.add_arguments("--batch_size", type=int)
-    #parser.add_arguments("--seq_length", type=int)
-    #parser.add_arguments("--epochs", type=int)
-    parser.add_arguments("--learning_rate", type=float)
+    parser.add_argument("--data_path")
+    parser.add_argument("--output_path")
+    parser.add_argument("--batch_size", type=int)
+    #parser.add_argument("--seq_length", type=int)
+    #parser.add_argument("--epochs", type=int)
+    parser.add_argument("--learning_rate", type=float)
     args = parser.parse_args()
 
     return args