问题描述
如下代码使用在使用命令行启动时不会出现问题,但是在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)
解决方法
由于在jupyter notebook中,args不为空。
将上面代码中的最后一行修改如下即可
args = parser.parse_args(args=[])
问题得到解决~