Skip to content

Instantly share code, notes, and snippets.

@edumunozsala
Created November 22, 2019 18:22
Show Gist options
  • Save edumunozsala/769549a729541a211a7c4a71f9da2988 to your computer and use it in GitHub Desktop.
Save edumunozsala/769549a729541a211a7c4a71f9da2988 to your computer and use it in GitHub Desktop.
Parser of arguments to the model
parser = argparse.ArgumentParser()
parser.add_argument('--data-folder', type=str, dest='data_folder', help='data folder mounting point')
parser.add_argument('--batch-size', type=int, dest='batch_size', default=50, help='mini batch size for training')
parser.add_argument('--x_filename', type=str, dest='x_filename', help='Filename with training data')
parser.add_argument('--y_filename', type=str, dest='y_filename', help='Filename with label data')
parser.add_argument('--training_size', type=str, dest='training_size', help='Size of training dataset')
parser.add_argument('--n_epochs', type=int, dest='n_epochs', help='Number of epochs')
args = parser.parse_args()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment