본문 바로가기

캡스톤디자인

딥러닝 모델: TFT, DeepAR, N-BEATS

1. TFT & 2. DeepAR

두 모델을 같은 코드 블럭 내에서 돌리려는데

TFT모델에서 자꾸 아래와 같은 오류가 발생했음.

 

TypeError: `model` must be a `LightningModule` or `torch._dynamo.OptimizedModule`, got `TemporalFusionTransformer`

 

맥북 로컬에서 돌려서 버전 충돌나는건가 싶어서 코랩으로 옮겨서 다시 돌려봤는데도 같은 오류...

챗지피티가 해결해주지 못함

 

결국 구글링을 해보고 Therefore, all you need to do is to replace 'pytorch_lightning' with 'lightning.pytorch' in the your code.  라는 스택오버플로우의 글을 발견해서 이대로 수정함

import pytorch_lightning as pl 대신 import lightning.pytorch as pl 을 사용함

그랬더니 허무하게도 문제가 해결됨...나는 뭐를 위해 며칠을 삽질했는가

 

그랬는데 이제는 RAM이 터짐 

코랩 프로인데 에포크 한개도 못돌리고 터지는게 맞나?

 

모델을 바꿔봐야겠으여...

 

3. N-BEATS

- 오로지 시간과 target변수만을 가지고 훈련하는 모델이라고 해서 기각

- 우리 프로젝트에서는 다양한 외부변수도 써야하는데 이 모델이 그걸 반영하지 못할 것 같음