연구자용 —
최신의 연구들을 찾아보고 모델을 확장하세요.
사용 방법 —
모델 공개하기
파이토치 허브는
hubconf.py
파일을 간단히 추가하여 사전 학습된 모델(모델 정의 및 사전 학습된 가중치)을 GitHub 저장소에 게시하는 것을 지원합니다.
모델 불러오기
사용자는
torch.hub.load()
API를 사용하여 사전 학습된 모델을 사용할 수 있습니다.
다음은 pytorch/vision
저장소에서 resnet18
을 불러오는 예시입니다.
model = torch.hub.load('pytorch/vision', 'resnet18', pretrained=True)