Spaces:
Running
Running
update
Browse files- examples/dfnet/step_2_train_model.py +2 -2
- requirements.txt +1 -1
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=
|
41 |
-
parser.add_argument("--patience", default=
|
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
|
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
|