group.add_argument("--required-batch-size-multiple", default=8, type=int, metavar="N",
help="batch size will be a multiplier of this value")
parser.add_argument("--dataset-impl", metavar="FORMAT", help="output dataset implementation",
choices=["raw", "lazy", "cached", "mmap"], default="cached")
if train:
group.add_argument("--train-subset", default="train", metavar="SPLIT",
choices=["train", "valid", "test"],
After Change
group.add_argument("--required-batch-size-multiple", default=8, type=int, metavar="N",
help="batch size will be a multiplier of this value")
parser.add_argument("--dataset-impl", metavar="FORMAT",
choices=get_available_dataset_impl(),
help="output dataset implementation")
if train:
group.add_argument("--train-subset", default="train", metavar="SPLIT",
