karfly
11/15/2017 - 10:27 AM

argparse example

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)