当我通过代码运行Addernet时,由它自己的文件提供 源代码:https://github.com/huawei-noah/AdderNet ,我在使用args时遇到了“系统退出2”,下面是我遇到该错误的一段代码
import os
from resnet20 import resnet20
import torch
from torch.autograd import Variable
from torchvision.datasets import CIFAR10
import torchvision.transforms as transforms
from torch.utils.data import DataLoader
import argparse
import math
parser = argparse.ArgumentParser(description='train-addernet')
# Basic model parameters.
parser.add_argument('--data', type=str, default='/cache/data/')
parser.add_argument('--output_dir', type=str, default='/cache/models/')
args = parser.parse_args()
os.makedirs(args.output_dir, exist_ok=True)
非常感谢
目前没有回答
相关问题 更多 >
编程相关推荐