연구자용 —

최신의 연구들을 찾아보고 모델을 확장하세요.


사용 방법 —

모델 공개하기

파이토치 허브는 hubconf.py 파일을 간단히 추가하여 사전 학습된 모델(모델 정의 및 사전 학습된 가중치)을 GitHub 저장소에 게시하는 것을 지원합니다.

모델 불러오기

사용자는 torch.hub.load() API를 사용하여 사전 학습된 모델을 사용할 수 있습니다.

다음은 pytorch/vision 저장소에서 resnet18을 불러오는 예시입니다.

model = torch.hub.load('pytorch/vision', 'resnet18', pretrained=True)