[FEAT] 모델 학습 코드 pytorch-template 포팅

[2018007956]

Background

• 더 좋은 코드에 대한 고민
• 모듈화가 잘 되어있는 pytorch-template을 가져와서 사용하고자 함
• 관리 용이성, 체계적이고 생산적인 모델 학습 환경 구축, 하드 코딩 시간 단축을 위함
• 고민이 되는 부분
  • MLflow에서 자동으로 세세한 로깅을 해주는 Autolog는 pytorch.nn.Module을 지원하지 않음
  • Pytorch Lightning은 지원함
  • 이 기능을 사용하기 위해 pytorch-lightning template을 사용해 보는 건 어떨까?

To do

• [x] pytorch-lightning template에 대한 고민
• [x] 선정된 template을 우리의 목적에 맞게 수정하여 사용

[2018007956]

모델 학습 보다 서비스 배포 단계까지의 구현(프론트 및 백엔드 개발)에 더욱 힘을 쏟고 있는 상황이기에 시간 상 pytorch-lightning 변환 작업은 진행하지 않기로 하였습니다. 모델 학습 시 로깅이 필요한 값들을 수동으로 지정해 주어야 합니다.

[2018007956]

같은 작업을 하는 이슈가 #86 에 생성되었습니다. PR #87 #89 참고부탁드립니다.