diff --git a/funasr/bin/train.py b/funasr/bin/train.py
index 22387b993..56a9269b7 100755
--- a/funasr/bin/train.py
+++ b/funasr/bin/train.py
@@ -160,6 +160,7 @@ def get_parser():
     )
     parser.add_argument(
         "--patience",
+        type=int_or_none,
         default=None,
         help="Number of epochs to wait without improvement "
              "before stopping the training",