빠르고 간편한 Pytorch Lightning 개념 정리

깃헙에서 코드를 구경하다가 Pytorch Lightning 을 발견했는데요. (공식 홈페이지 ) 기존 Pytorch 보다 코드가 훨씬 단축되어서 딥러닝 작업을 훨씬 간단히 수행할 수 있습니다. Pytorch와 API도 유사해서 기존 Pytorch 코드도 쉽게 변환이 가능합니다. 가장 좋은 점은 기존에 Pytorch에서 했던 데이터 로더, 모델 정의,손실함수, 최적화 및 학습 루프 등 딥러닝 작업을 위한 기본 모델 구성을 , Python Lightning에서는 모듈화한 간단한 명령으로 수행이 가능합니다. Pytorch Lightning 설치 pip install pytorch-lightning Pytorch와 Pytorh_lightning 비교 Pytorch # PyTorch import torch imp..

원문링크 : 빠르고 간편한 Pytorch Lightning 개념 정리