목록pyTorch (1)
핀아의 저장소 ( •̀ ω •́ )✧
[CRAFT-pytorch] ImportError: cannot import name 'model_urls' from 'torchvision.models.vgg'
OCR 모델을 위한 환경 셋팅 중 직면한 문제. 원인은 torchvision 버전이 업그레이드가 되면서 'model_urls'가 사라졌다. 따라서 해결 방법은 간단하다. 'model_urls' 부분을 삭제해주는 것! 1. vgg16_bn.py에 들어간다. 2. 'model_urls' 부분을 삭제한다. from torchvision import models from torchvision.models.vgg import model_urls
Code & Issues/Issue
2023. 8. 16. 11:46