save 한 모델 적용해서 이미지 분류값 예측하기!(pth 파일)

베스트 결과가 나온 모델을 저장해놓고 사용하는 법을 간략히 소개합니다! Resnet18 예시) model = models.resnet18(pretrained=True) model.load_state_dict(torch.load('base_model.pth', map_location=torch.device('cpu'))) transform = transforms.Compose([ transforms.RandomHorizontalFlip(p=0.5), transforms.RandomVerticalFlip(p=0.5), transforms.Resize((224, 224)), transforms.ToTensor(), transforms.Normalize((0.5,), (0.5,)) ]) 먼저 어떤 모델을 사용할.....