当我使用 jupyter notebook 训练模型时，“ipykernel_launcher.py: error: unrecognized arguments”存在

Question

我正在学习使用 jupyter notebook 来使用 Pytorch 训练模型。但是当我运行单元格时，这里出现了一个问题。我该怎么办？ 这是我认为可能存在问题的代码。

parser = argparse.ArgumentParser(description='PyTorch Training')
parser.add_argument('--batch_size', default=1, type=int, metavar='BT',
                    help='batch size')
# =============== optimizer
parser.add_argument('--lr', '--learning_rate', default=1e-6, type=float,
                    metavar='LR', help='initial learning rate')
parser.add_argument('--momentum', default=0.9, type=float, metavar='M',
                    help='momentum')
parser.add_argument('--weight_decay', '--wd', default=2e-4, type=float,
                    metavar='W', help='default weight decay')
parser.add_argument('--stepsize', default=3, type=int, 
                    metavar='SS', help='learning rate step size')
parser.add_argument('--gamma', '--gm', default=0.1, type=float,
                    help='learning rate decay parameter: Gamma')
parser.add_argument('--maxepoch', default=30, type=int, metavar='N',
                    help='number of total epochs to run')
parser.add_argument('--itersize', default=10, type=int,
                    metavar='IS', help='iter size')
# =============== misc
parser.add_argument('--start_epoch', default=0, type=int, metavar='N',
                    help='manual epoch number (useful on restarts)')
parser.add_argument('--print_freq', '-p', default=1000, type=int,
                    metavar='N', help='print frequency (default: 50)')
parser.add_argument('--gpu', default='0', type=str,
                    help='GPU ID')
parser.add_argument('--resume', default='', type=str, metavar='PATH',
                    help='path to latest checkpoint (default: none)')
parser.add_argument('--tmp', help='tmp folder', default='tmp/RCF')
# ================ dataset
parser.add_argument('--dataset', help='root folder of dataset', default='data/HED-BSDS_PASCAL')
args = parser.parse_args()

错误是：

"usage: ipykernel_launcher.py [-h] [--batch_size BT] [--lr LR] [--momentum M]
                             [--weight_decay W] [--stepsize SS]
                             [--gamma GAMMA] [--maxepoch N] [--itersize IS]
                             [--start_epoch N] [--print_freq N] [--gpu GPU]
                             [--resume PATH] [--tmp TMP] [--dataset DATASET]
ipykernel_launcher.py: error: unrecognized arguments: -f C:\Users\59863\AppData\Roaming\jupyter\runtime\kernel-4a7fc5df-ea52-4c9e-93a9-f466074bfc90.json"

Answer 1

ArgumentParser.parse_args(args=None, namespace=None) 将args（参数字符串列表）转换为对象并将它们分配为命名空间的属性。

如果未提供args，则默认取自sys.argv。在您的情况下，sys.argv 是用于启动 jupyter notebook 的参数。

所以如果你想在 jupyter notebook 中使用ArgumentParser，请手动传入args：

args = parser.parse_args(['--batch_size', '32']）