251001 [ Day 62 ] - Project (7)

시작하며

오늘은 새롭게 작업한 API와 AI 모델 파이프라인을 연동시켜서 파이프라인의 response 값을 API로 받아 웹에 출력시키는 것까지 테스트하였다.

모델 학습

이후 모든 데이터를 사용한 본격적인 모델 학습을 시작하였다.

Data

• 모든 클래스 합쳐서 약 7만장의 이미지 데이터

Model

• ResNet18

Epoch

• 15

Learning Rate

• 0.0001

Batch Size

• 64

마치며

수업이 끝나고 바로 학습을 시켰는데 5시간이 지났지만 아직 절반도 안 끝났다. 중간에 정확도와 손실값을 보고 어느 정도 학습했다는 판단이 들면 중단하고 연휴 동안 파인 튜닝을 하면서 학습시켜봐야겠다.