Pytorch Lightning

pytorch lightning(PL) hooks 최근 두 달 pytorch lightning(as PL)을 찍먹하며 쓸수록 괜찮은 프레임워크라는 생각이든다. 먼저 PL은 pytorch, tensorflow(as TF), onnx와 같은 또 다른 프레임워크가 아니다

이번 글에서는 Pytorch Lightning(as PL)의 두가지 핵심이라 생각되는 파트중 하나인 LightningModule (as PL.LM) 에서 training loop를 구성하는 내용에 대해 알아보려 한다. PL를 처음 사용 하면서 가장 남감 했던 부분