import argparse
parser = argparse.ArgumentParser()
parser.add_argument('--train-dataset-path', type=str,
help='path to train dataset')
parser.add_argument('--val-dataset-path', type=str,
help='path to val dataset')
parser.add_argument('--batch-size', type=int, default=32,
help='batch size')
parser.add_argument('--epochs', type=int, default=99999,
help='number of training epochs')
args = parser.parse_args()
print(args.val_dataset_path)