공부/그 외 썸네일형 리스트형 [3년차 텐서플로우 이용자의 pytorch 사용기] 커스텀 데이터 셋 사용하기 이전에 졸업 작품과 다른 연구에서 Eelectra 모델과 BERT를 이용하려고 pytorch를 사용했던 적이 있었는데 그때는 단일 모델에 CSV에 저장된 데이터를 사용했기 때문에 큰 어려운은 없었다 하지만 이번에 연구를 위해 CNN-LSTM 기반 모델을 사용하는데 데이터를 메모리에 로드하는 과정이 추가된것 코드를 간단하게 작성할 수 있는 파이토치 라이트닝을 사용할 예정이다. 일반 파이토치는 훈련과정을 에포크 for문을 통해 작성을해야해서 여간 귀찮은게 아닌... 일단 데이터셋 클래스를 작성해준다.import numpy as npfrom keras.preprocessing.image import img_to_array, load_imgfrom keras.applications.mobilenet_v2 i.. 더보기 이전 1 다음
