efficientnet v2 다운로드, 사용

download 링크는 발견했는데 pth파일을 어떻게 열어야 할지 모르겠네요.
https://download.pytorch.org/models/efficientnet_v2_s-dd5fe13b.pth
torchhub를 이용해야되는것 같긴한데…

안녕하세요~
PyTorch에서는 모델을 저장할 때 .pt 또는 .pth 확장자를 사용하는 것이 일반적인 규칙입니다.
해당 파일 사용과 관련된 튜토리얼 페이지 남겨드립니다.

좋아요 1

감사합니다~~

URL의 생김새(?)가 PyTorch Model Hub에서 제공하는 것 같습니다.

아래 문서를 읽어보시면 PyTorch에서 공식으로 제공하고 있는 학습된 모델(pre-trained model)들을 불러오는 방법을 확인하실 수 있습니다.

https://pytorch.org/vision/stable/models.html

더불어, torchvision.models.efficientnet_v2_s 부분도 한 번 살펴보시면 좋을 것 같습니다. :slight_smile:

https://pytorch.org/vision/stable/models/generated/torchvision.models.efficientnet_v2_s.html#torchvision.models.efficientnet_v2_s

감사합니다!!

좋아요 1

이 글은 마지막 댓글이 달린지 오래(30일)되어 자동으로 닫혔습니다. 댓글 대신 새로운 글을 작성해주세요! :slight_smile: