[PyTorch] 기존 제공 model 불러오기

- torchvision.models를 통해 model 불러오기 

import torchvision.models as models

 

- model 설정 (image clf : resnet34, vgg, alexnet, squeezenet, densenet, googlenet etc)

model = models.resnet34(pretrained=False)

 

- model.cuda = model.to(DEVICE) 모델 입력

model = model.cuda # model.to(DEVICE)