chenyuntc / pytorch-book

PyTorch tutorials and fun projects including neural talk, neural style, poem writing, anime generation (《深度学习框架PyTorch:入门与实战》)
MIT License
11.99k stars 3.73k forks source link

作者你好,我在运行第六章猫狗分类程序一直报错No module named 'resnet34' #220

Closed halfsummer583 closed 4 years ago

halfsummer583 commented 4 years ago

coding:utf-8

from config import opt import os import torch as t

import sys sys.path.append("E:/desktop project/pytorch-code/chapter06-best_practice/models/resnet34.py") from models.resnet34 import ResNet34

from data.dataset import DogCat from torch.utils.data import DataLoader from torchnet import meter from utils.visualize import Visualizer from tqdm import tqdm

报错: (base) E:\desktop project\pytorch-code\chapter06-best_practice>python main.py train --train-data-root=./data/train --use-gpu --env=classifier Traceback (most recent call last): File "main.py", line 8, in from models.resnet34 import ResNet34 File "E:\desktop project\pytorch-code\chapter06-best_practice\models__init__.py", line 2, in from resnet34 import ResNet34 ModuleNotFoundError: No module named 'resnet34' 图片