timm, pytorch-lighting을 사용한 코드입니다.
원래 하나의 파이썬 파일로 작성한걸 노트북에 붙여넣어서 이런 모양이 되었습니다.

이 코드는 모델 파일의 크기가 100MB 정도의 작은 모델을 사용했습니다.
배치사이즈 32에서 VRAM 사용량은 8GB가 조금 안됩니다.
아마 모델을 큰 모델로 바꾸는것만으로도 점수가 더 올라가지 않을까 생각합니다.

코드 맨 밑에는 학습한 모델을 불러오는 방법을 적어두었으니 도움이 되길 바랍니다. 감사합니다.