AttributeError: Can't get attribute 'BERTClassifier' on module '__main__' from 'path'

Error

• Flask에서 model을 불러오려고 하니 다음과 같이 error가 발생하였다.

    sentimentModel = torch.load('./model.pt', map_location='cpu')
  

  

Error 해결

• torch.save(model, 'model.pt')로 저장하여 model 전체를 불러오려고 하니 어떤 모델인지 정의 되어 있지 않아 에러가 나는거 같았다.

• torch.save(model.state_dict(), 'model.pt')로 저장한 후 모델을 정의하고 정의된 모델에 load를 하여 가중치와 편향을 부여하는 방법으로 해결하였다.

    model = BERTClassifier(bertmodel,  dr_rate=0.5).to(device)
      
    model.load_state_dict(torch.load('model_state_dict.pt', map_location='cpu'))
  

참고 사이트

[Pytorch] 모델 저장 방법, 그리고 전체 저장과 state_dict 저장의 차이