search

yanx27 / EverybodyDanceNow_reproduce_pytorch

Everybody dance now reproduced in pytorch
MIT License
605 stars 173 forks source link

run train_pose2vid.py，it shows ModuleNotFoundError: No module named 'data.data_loader' #59

Open wangzheliu opened 5 years ago

wangzheliu commented 5 years ago

(torch4) [liu@no3@node04 EverybodyDanceNow_reproduce_pytorch]$ python train_pose2vid.py Traceback (most recent call last): File "train_pose2vid.py", line 17, in from data.data_loader import CreateDataLoader ModuleNotFoundError: No module named 'data.data_loader' how to fix it？

sandhyalaxmiK commented 5 years ago

create empty init.py file in data folder

disanda commented 4 years ago

it is in main file , dir is ‘.\src\pix2pixHD\data’ , and there are a file named 'data_loader.py' .it is what your need.do you find it?

maveeshah commented 4 years ago

(torch4) [liu@no3@node04 EverybodyDanceNow_reproduce_pytorch]$ python train_pose2vid.py Traceback (most recent call last): File "train_pose2vid.py", line 17, in from data.data_loader import CreateDataLoader ModuleNotFoundError: No module named 'data.data_loader' how to fix it？

previous code `pix2pixhd_dir = Path('./src/pix2pixHD/') sys.path.append(str(pix2pixhd_dir))

from data.data_loader import CreateDataLoader from models.models import create_model import util.util as util from util.visualizer import Visualizer from util import html import src.config.test_opt as opt`

change it to `pix2pixhd_dir = Path('./src/') sys.path.append(str(pix2pixhd_dir))

from pix2pixHD.data.data_loader import CreateDataLoader from pix2pixHD.models.models import create_model import pix2pixHD.util.util as util from pix2pixHD.util.visualizer import Visualizer from pix2pixHD.util import html import config.test_opt as opt`

you'll have to do it in many files and it'll work smoothly then