import argparse
def parse_args():
parser = argparse.ArgumentParser(description="Pytorch implementation of GAN models.")
parser.add_argument('--is_train', type=bool, default=True)
parser.add_argument('--dataroot', type=str, default='cifar')
parser.add_argument('--load_G', type=bool, default=False, help='Path for loading Generator network')
args = parser.parse_args()
return args
上面的代码会以下报错:
usage: ipykernel_launcher.py [-h] [--is_train IS_TRAIN] [--dataroot DATAROOT]
[--load_G LOAD_G]
ipykernel_launcher.py: error: unrecognized arguments: -f C:\Users\ASUS\AppData\Roaming\jupyter\runtime\kernel-e34db78d-3d1e-48f4-8f76-19ec92502171.json
An exception has occurred, use %tb to see the full traceback.
SystemExit: 2
解决方案是:
import argparse
def parse_args():
parser = argparse.ArgumentParser(description="Pytorch implementation of GAN models.")
parser.add_argument('--is_train', type=bool, default=True)
parser.add_argument('--dataroot', type=str, default='cifar')
parser.add_argument('--load_G', type=bool, default=False, help='Path for loading Generator network')
args = parser.parse_args([])
return args