leecs0503 / NLP_FinalProject

FinalProject
1 stars 3 forks source link

Feature/8 implement lstm model #17

Closed leecs0503 closed 2 years ago

leecs0503 commented 2 years ago

8

구현내용: ImageChannel: pretrain된 vgg19모델 활용 TextChannel: word embedding과 lstm 활용 LSTM_VQA: 두 체널을 활용해 vqa에 맞도록 학습되도록 적용

테스트코드: 각 imagechannel에 맞는 크기의 tensor을 넣어줬을 때 적절한 크기의 tensor가 반환되는지 확인