Skip to content

fix No module named 'torchvision.models.utils' error#31

Open
pgchui wants to merge 2 commits intoTuroad:mainfrom
pgchui:main
Open

fix No module named 'torchvision.models.utils' error#31
pgchui wants to merge 2 commits intoTuroad:mainfrom
pgchui:main

Conversation

@pgchui
Copy link

@pgchui pgchui commented Dec 13, 2021

Error when running detect.py

Error message:

  File "<directory_to_lanedet>/lanedet/lanedet/models/backbones/mobilenet.py", line 4, in <module>
    from torchvision.models.utils import load_state_dict_from_url
ModuleNotFoundError: No module named 'torchvision.models.utils'

torchvision verison = 0.11.1

Using from torch.hub import load_state_dict_from_url fixed the problem.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants