문제
RuntimeError: invalid hash value (expected "3dd342df", got "7f5810bc96def8f7552d5b7e68d53c4786f81167d28291b21c0d90e1fca14934")
문제 설명
torchvision 버전 오류, 해시 값 불일치 문제
해시는 파일의 무결성을 확인하기 위한 방식
해결 방법
1.
EDIT: 이 문제는 torchvision 0.17에서 해결
•
"torchvision>0.16"을 설치하면 해결된다.
2.
근데 만약 0.16에서 막혔다면 아래에 설명된 대로 하십쇼.
from torchvision.models import efficientnet_b0, EfficientNet_B0_Weights
from torchvision.models._api import WeightsEnum
from torch.hub import load_state_dict_from_url
def get_state_dict(self, *args, **kwargs):
kwargs.pop("check_hash")
return load_state_dict_from_url(self.url, *args, **kwargs)
WeightsEnum.get_state_dict = get_state_dict
efficientnet_b0(weights=EfficientNet_B0_Weights.IMAGENET1K_V1)
efficientnet_b0(weights="DEFAULT")
SQL
복사