如何在pytorch中实现parse_args（）？

Question

我正在尝试为我的深度学习模型编码一个模块。我希望使用ArgeParse存储这些论点。 args = parser.parse_args（）中存在一些问题！另外，使用ArgParse库有什么好处？

import numpy as np
import argparse
    
if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument('--dataset', type=str,
                        help='Dataset')
    parser.add_argument('--epoch', type=int, default=40,
                        help='Training Epochs')
    parser.add_argument('--node_dim', type=int, default=64,
                        help='Node dimension')
    parser.add_argument('--num_channels', type=int, default=2,
                        help='number of channels')
    parser.add_argument('--lr', type=float, default=0.005,
                        help='learning rate')
    parser.add_argument('--weight_decay', type=float, default=0.001,
                        help='l2 reg')
    parser.add_argument('--num_layers', type=int, default=2,
                        help='number of layer')
    parser.add_argument('--norm', type=str, default='true',
                        help='normalization')
    parser.add_argument('--adaptive_lr', type=str, default='false',
                        help='adaptive learning rate')

    args = parser.parse_args()
    
    print(args)

上面的代码是完整代码的一部分，如下面的链接

错误：

usage: ipykernel_launcher.py [-h] [--dataset DATASET] [--epoch EPOCH]
                             [--node_dim NODE_DIM]
                             [--num_channels NUM_CHANNELS] [--lr LR]
                             [--weight_decay WEIGHT_DECAY]
                             [--num_layers NUM_LAYERS] [--norm NORM]
                             [--adaptive_lr ADAPTIVE_LR]
ipykernel_launcher.py: error: unrecognized arguments: -f /Users/anshumansinha/Library/Jupyter/runtime/kernel-1e4c0f41-a4d7-4388-be24-640dd3411f56.json
An exception has occurred, use %tb to see the full traceback.

SystemExit: 2

上述代码是代码的一部分，可以在以下github链接：链接

Answer 1

您使用参数-f称为ipykernel_launcher.py。但是-f不在您的参数列表中。
-f是参数，您要用于输入文件吗？然后，您应该将类​​似的内容添加到您的代码中：

parser.add_argument('--file', '-f', type=str)

ArgParse的好处是，您不必手工编写代码来解析参数。您可以尝试一下。它比通常想象的更多的代码。特别是如果您有位置论点。