jupyer notebook使用argparse报错问题解决方法

问题描述

如下代码使用在使用命令行启动时不会出现问题,但是在jupyter notebook执行会出现报错,如图所示

import argparse
parser = argparse.ArgumentParser(description='PyTorch implementation of Deep GCN')

# base
parser.add_argument('--phase', default='train', type=str, help='train or test(default)')
parser.add_argument('--use_cpu', action='store_true', help='use cpu?')

# dataset args
parser.add_argument('--data_dir', type=str, default='data/ppi')
parser.add_argument('--batch_size', default=1, type=int, help='mini-batch size (default:8)')
parser.add_argument('--in_channels', default=50, type=int, help='the channel size of input point cloud ')

args = parser.parse_args()
print(args)

jupyer notebook使用argparse报错问题解决方法_第1张图片

解决方法

由于在jupyter notebook中,args不为空。

将上面代码中的最后一行修改如下即可

args = parser.parse_args(args=[])

jupyer notebook使用argparse报错问题解决方法_第2张图片

问题得到解决~

你可能感兴趣的:(python)