torchlightning

인공지능공부/프레임워크

pytorch lightning을 사용해보자

pytorch는 가장 널리 쓰이는 딥러닝 프레임워크중 하나이죠. 딥러닝, 머신러닝을 공부하시는 분이라면 들어보셨을겁니다. 써보신 분들은 아시겠지만 pytorch 자체가 굉장히 자유도가 높은 프레임워크이죠. 일반적으로 딥러닝 모델을 구현할 경우 크게 복잡하지는 않지만 GPU, 분산 처리 등을 구현할 경우 코드가 조금씩 복잡해집니다. pytorch lightning은 굉장히 잘 정돈된 하나의 틀을 제공해줍으로써 pytorch의 가독성을 높여줍니다. 이제 본격적으로 pytorch lightning에 대해서 알아보도록 하죠. pytorch lightning 사용 이유 1. 간단한 학습, 검증과정 pytorch lightning은 pytorch의 학습, 검증 과정을 training_step, validation_..

컴공누나
'torchlightning' 태그의 글 목록