This commit is contained in:
嘉渊 2023-04-25 01:02:28 +08:00
parent fc68b4c315
commit 52f910aae2

View File

@ -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,