diff --git a/scripts/train.py b/scripts/train.py index 829e670..5c9b136 100644 --- a/scripts/train.py +++ b/scripts/train.py @@ -60,7 +60,7 @@ def get_args(): ) parser.add_argument( - '-p', + '-t', '--patience', default=5, type=int,