hvd_sm_commandline.py

# Change 8: Update script to accept hyperparameters as command line arguments
parser = argparse.ArgumentParser()

# Hyper-parameters
parser.add_argument('--epochs',        type=int,   default=15)
parser.add_argument('--learning-rate', type=float, default=0.001)
parser.add_argument('--batch-size',    type=int,   default=256)
parser.add_argument('--weight-decay',  type=float, default=2e-4)
parser.add_argument('--momentum',      type=float, default='0.9')
parser.add_argument('--optimizer',     type=str,   default='adam')