def add_args(parser):
    parser.add_argument('--batch-size', default=32, type=int, help='Number of inputs in a mini-batch')
    parser.add_argument('--size', default=1024, type=int, help='Dense layer size')
    parser.add_argument('--train', action='store_true', dest='train', help='Compute loss and optimization pass')
    parser.add_argument('--include-activation', action='store_true', dest='activation', help='Include ReLU activation (otherwise linear/no activation')
    parser.set_defaults(train=False, batches_per_step=5000, steps=5)
    return parser
def add_args(parser):
    parser.add_argument('--batch-size', default=32, type=int, help='Number of inputs in a mini-batch')
    parser.add_argument('--size', default=1024, type=int, help='Dense layer size')
    parser.add_argument('--train', action='store_true', dest='train', help='Compute loss and optimization pass')
    parser.add_argument('--include-activation', action='store_true', dest='activation', help='Include ReLU activation (otherwise linear/no activation')
    parser.set_defaults(train=False, batches_per_step=5000, steps=5)
    return parser