search

VL-Group / PENET

[CVPR 2023]Official Pytorch code for paper "Prototype-based Embedding Network for Scene Graph Generation"
https://arxiv.org/abs/2303.07096
MIT License
46 stars 7 forks source link

AttributeError: module 'torchvision' has no attribute 'datasets' #12

Open CL2LC opened 4 weeks ago

CL2LC commented 4 weeks ago

哪位运行成功的朋友能告诉我，你们使用的pytorch以及torhvision的版本是什么y呀。我目前在cuda==10.1尝试了pytorch==1.13.1,torhvision==0.14.1以及 pytorch==1.11.0,torhvision==0.12.0两种，都报错AttributeError: module 'torchvision' has no attribute 'datasets'，GPU是3090。

Someone who has been successful can tell me what version of pytorch and torhvision you use. I am currently trying pytorch==1.13.1,torhvision==0.14.1 and cuda==10.1 pytorch==1.11.0,torhvision==0.12.0 Both report AttributeError: module 'torchvision' has no attribute 'datasets', GPU is 3090

Kaguya7711 commented 4 weeks ago

我用torch==1.7.0，torchvision==0.8.0不会报错