嘉渊 2 ani în urmă
părinte
comite
52f910aae2
1 a modificat fișierele cu 6 adăugiri și 0 ștergeri
  1. 6 0
      funasr/bin/train.py

+ 6 - 0
funasr/bin/train.py

@@ -297,6 +297,12 @@ def get_parser():
         default="small",
         help="whether to use dataloader for large dataset",
     )
+    parser.add_argument(
+        "--dataset_conf",
+        action=NestedDictAction,
+        default=dict(),
+        help=f"The keyword arguments for dataset",
+    )
     parser.add_argument(
         "--train_data_file",
         type=str,