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)

解决方法

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

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

args = parser.parse_args(args=[])

问题得到解决~