argparse中add_argument()的用法

https://docs.python.org/zh-cn/3/library/argparse.html

import argparse

parser = argparse.ArgumentParser()

parser.add_argument('--dataset-type', type=str, default='parts3_2', choices=['parts3_2', 'stomach2'], help='type of the dataset')

parser.add_argument("--dataset-path", default='./datasets1', type=str,  help="Path of the trainset.")

parser.add_argument('--training-set-ratio', type=float, default=1.0, help='ratio of the training set to use')

parser.add_argument('--batch-size', type=int, default=64, help='batch size')

parser.add_argument('--lr', type=float, default=1e-3, help='learning rate')

parser.add_argument("--lr-gamma", type=float, default=0.1, help='gamma of lr scheduler')

parser.add_argument('--weight-decay', type=float, default=1e-4, help='weight decay')

parser.add_argument('--dropout', type=float, default=0.7, help='dropout ratio')
parser.add_argument('--epochs', type=int, default=20, help='number of epochs')

parser.add_argument('--seed', type=int, default=666, help='seed')

parser.add_argument('--num-workers', type=int, default=0, help='number of workers for the dataloaders')

parser.add_argument('--checkpoint-root', type=str, default='./checkpoint', help='path to the checkpoint')

parser.add_argument('--model-name', type=str, default='ResNet50', help='name of the model')

args = parser.parse_args()

# In [11]: args
# Out[11]: Namespace(batch_size=64, checkpoint_root='./checkpoint', dataset_path='./datasets1', dataset_type='parts3_2', dropout=0.7, epochs=20, lr=0.001, lr_gamma=0.1, model_name='ResNet50', num_workers=0, seed=666, training_set_ratio=1.0, weight_decay=0.0001)

# In [12]: args.training_set_ratio
# Out[12]: 1.0

# In [13]: args.batch_size
# Out[13]: 64

# In [14]: args.checkpoint_root
# Out[14]: './checkpoint'

print(args.dataset_type)
print(args.training_set_ratio)