嘉渊 hace 2 años
padre
commit
decdbc3c97
Se han modificado 1 ficheros con 1 adiciones y 0 borrados
  1. 1 0
      funasr/bin/train.py

+ 1 - 0
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",