search

NVIDIA / pix2pixHD

Synthesizing and manipulating 2048x1024 images with conditional GANs
https://tcwang0509.github.io/pix2pixHD/
Other
6.62k stars 1.39k forks source link

not work in python 3.10 #335

Open kingstone101 opened 6 months ago

kingstone101 commented 6 months ago

CustomDatasetDataLoader Traceback (most recent call last): File "C:\Users\GAMER\Desktop\pix2pixHD-master\train.py", line 43, in data_loader = CreateDataLoader(opt) File "C:\Users\GAMER\Desktop\pix2pixHD-master\data\data_loader.py", line 6, in CreateDataLoader data_loader.initialize(opt) File "C:\Users\GAMER\Desktop\pix2pixHD-master\data\custom_dataset_data_loader.py", line 20, in initialize self.dataset = CreateDataset(opt) File "C:\Users\GAMER\Desktop\pix2pixHD-master\data\custom_dataset_data_loader.py", line 7, in CreateDataset from data.aligned_dataset import AlignedDataset File "C:\Users\GAMER\Desktop\pix2pixHD-master\data\aligned_dataset.py", line 2, in from data.base_dataset import BaseDataset, get_params, get_transform, normalize File "C:\Users\GAMER\Desktop\pix2pixHD-master\data\base_dataset.py", line 3, in import torchvision.transforms as transforms File "C:\Users\GAMER\AppData\Local\Programs\Python\Python310\lib\site-packages\torchvision__init__.py", line 6, in from torchvision import _meta_registrations, datasets, io, models, ops, transforms, utils File "C:\Users\GAMER\AppData\Local\Programs\Python\Python310\lib\site-packages\torchvision_meta_registrations.py", line 163, in @torch._custom_ops.impl_abstract("torchvision::nms") AttributeError: module 'torch._custom_ops' has no attribute 'impl_abstract'