argument_parser.py

import argparse

def parse_arguments():
    parser = argparse.ArgumentParser()

    # Attack settings.
    # parser.add_argument('', default=, type=, choices=[])
    parser.add_argument('--threshold', default = 0.05, type=float)
    parser.add_argument('--ws_defense', action='store_true')
    parser.add_argument('--conv', action='store_true')

    parser.add_argument('--sigma', default=1e-4, type=float)
    parser.add_argument('--lb', default=2.5, type=float)

    # Model settings.
    parser.add_argument('--model', default='VGG16', type=str, choices=['VGG16', 'resnet18'])
    parser.add_argument('--load', action='store_true')
    parser.add_argument('--save', action='store_true')
    parser.add_argument('--max_epoch', default = 100, type=int)
    parser.add_argument('--learning_rate', default = 0.1, type=float)

    # Data settings.
    parser.add_argument('--dataset', default='CIFAR10', type=str, choices=['MNIST','CIFAR10','GTSRB','TinyImageNet'])
    parser.add_argument('--batch_size', default = 512, type=int)

    args = parser.parse_args()
    return args