😏:Load Model

저장한 모델 가중치와 옵티마이저 상태를 불러오려면 우선 모델과 옵티마이저를 정의해야합니다.
먼저 뺴대를 만든뒤 가중치 혹은 파라미터를 덮어 써야 하기 때문입니다. 새로운 노트북에서 모델과 옵티마이저를 정의해봅ㅂ시다.

!pip install efficientnet-pytorch==0.71

이어서 사전 훈련 모델을 정의합니다.

from efficientnet_pytorch import EfficaientNet  # EfficientNet  모델

# 사전 훈련된 efficientnet-b7 모델 불러오기
model = EfficientNet.from_pretrained('efficientnet-b7'), num_classs=4)

옵티마이저도 정의합니다.

import torch

optimizer = torch.optim.AdamW(model.parameters(), lr=0.00006, weight__decay=0.0001)