在Google Colab上运行时,我得到了“系统出口2”

2024-10-02 12:32:23 发布

您现在位置:Python中文网/ 问答频道 /正文

当我通过代码运行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)  

非常感谢


Tags: fromimportaddparserdataostypeargparse

热门问题