HoneyTian commited on
Commit
66bf794
·
1 Parent(s): 69fa971
examples/dfnet/step_2_train_model.py CHANGED
@@ -37,8 +37,8 @@ def get_args():
37
  parser.add_argument("--train_dataset", default="train.xlsx", type=str)
38
  parser.add_argument("--valid_dataset", default="valid.xlsx", type=str)
39
 
40
- parser.add_argument("--num_serialized_models_to_keep", default=10, type=int)
41
- parser.add_argument("--patience", default=5, type=int)
42
  parser.add_argument("--serialization_dir", default="serialization_dir", type=str)
43
 
44
  parser.add_argument("--config_file", default="config.yaml", type=str)
 
37
  parser.add_argument("--train_dataset", default="train.xlsx", type=str)
38
  parser.add_argument("--valid_dataset", default="valid.xlsx", type=str)
39
 
40
+ parser.add_argument("--num_serialized_models_to_keep", default=15, type=int)
41
+ parser.add_argument("--patience", default=10, type=int)
42
  parser.add_argument("--serialization_dir", default="serialization_dir", type=str)
43
 
44
  parser.add_argument("--config_file", default="config.yaml", type=str)
requirements.txt CHANGED
@@ -1,4 +1,4 @@
1
- gradio==5.23.2
2
  datasets==3.2.0
3
  python-dotenv==1.0.1
4
  scipy==1.15.1
 
1
+ gradio
2
  datasets==3.2.0
3
  python-dotenv==1.0.1
4
  scipy==1.15.1