HoneyTian commited on
Commit
74d0273
·
1 Parent(s): 0e1473e
examples/frcrn/step_2_train_model.py CHANGED
@@ -42,8 +42,8 @@ def get_args():
42
  parser.add_argument("--train_dataset", default="train.xlsx", type=str)
43
  parser.add_argument("--valid_dataset", default="valid.xlsx", type=str)
44
 
45
- parser.add_argument("--num_serialized_models_to_keep", default=10, type=int)
46
- parser.add_argument("--patience", default=5, type=int)
47
  parser.add_argument("--serialization_dir", default="serialization_dir", type=str)
48
 
49
  parser.add_argument("--config_file", default="config.yaml", type=str)
 
42
  parser.add_argument("--train_dataset", default="train.xlsx", type=str)
43
  parser.add_argument("--valid_dataset", default="valid.xlsx", type=str)
44
 
45
+ parser.add_argument("--num_serialized_models_to_keep", default=15, type=int)
46
+ parser.add_argument("--patience", default=10, type=int)
47
  parser.add_argument("--serialization_dir", default="serialization_dir", type=str)
48
 
49
  parser.add_argument("--config_file", default="config.yaml", type=str)