목록AI/Pytorch_lightning (1)
대학원생의 그저 그런 기록
[Pytorch_lightning] Trainer에 대해 알아보자
PyTorch Lightning의 Trainer 클래스이번에는 파이토치 라이트닝에서 사용하는 Trainer 클래스에 대해 알아보려고 한다. 파이토치 라이트닝(Pytorch Lightning)의 Trainer는 딥러닝 모델 학습의 복잡한 과정을 간결하게 만들어 주는 핵심 클래스다. 이전 PyTorch는 모델을 학습할 때 데이터 준비, 모델 정의 및 여러가지 하이퍼 파라미터(optimizer, loop, check point 설정 등 여러가지)를 직접 설계해서 코딩을 해야했지만, Pytorch Lightning에서는 이러한 코드를 자동으로 처리해주기 때문에, 사용자는 모델의 구조와 데이터 처리 등 핵심적인 부분만 처리하면 된다. (생각보다 다른 것들을 신경쓰는데 오래걸렸는데 이 라이브러리를 통해 시간을 정말..
AI/Pytorch_lightning
2025. 6. 18. 02:36