Spaces:
Running
Running
update
Browse files
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=
|
46 |
-
parser.add_argument("--patience", default=
|
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)
|