模型复现之argparse参数解析

最新推荐文章于 2024-07-29 14:59:24 发布

小卜妞~

最新推荐文章于 2024-07-29 14:59:24 发布

阅读量798

点赞数 1

分类专栏： # 时空序列预测/智慧城市文章标签：深度学习

本文链接：https://blog.csdn.net/qq_33866063/article/details/115787718

版权

argparse

学习WaveNet代码时，遇到大量的argparse的使用，那么这个包究竟是什么用处呢？怎么使用呢？

import argparse

parser = argparse.ArgumentParser(description='PyTorch MNIST Example')
parser.add_argument('--epochs', type=int, default=10, metavar='N', help='number of epochs to train (default: 10)')
parser.add_argument('--lr', type=float, default=0.01, metavar='LR', help='learning rate (default: 0.01)')
parser.add_argument('--momentum', type=float, default=0.5, metavar='M', help='SGD momentum (default: 0.5)')
parser.add_argument('--no-cuda', action='store_true', default=False, help='disables CUDA training')
parser.add_argument('--seed', type=int, default=1, metavar='S', help='random seed (default: 1)')
parser.add_argument('--log-interval', type=int, default=10, metavar='N',
                    help='how many batches to wait before logging training status')
parser.add_argument('--batch-size', type=int, default=32, metavar='N',
                    help='input batch size for training (default: 32)')
parser.add_argument('--test-batch-size', type=int, default=32, metavar='N',
                    help='input batch size for testing (default: 32)')
args = parser.parse_args()
# ---------------------------结果---------------------------------------------
C:\Users\llrachel>python demo.py -h
usage: demo.py [-h] [--epochs N] [--lr LR] [--momentum M] [--no-cuda] # 程序名；所有属性名及其别名
               [-