diff --git a/scripts/train.py b/scripts/train.py index 5990d3c..829e670 100644 --- a/scripts/train.py +++ b/scripts/train.py @@ -62,7 +62,7 @@ def get_args(): parser.add_argument( '-p', '--patience', - default=8, + default=5, type=int, help="Patience for early stopping." )