반응형
딥러닝 모델 성능을 최적화 하기 위해 콜백 함수를 사용한다. 콜백 함수는 개발자가 명시적으로 함수를 호출하는 것이 아니라, 함수를 등록하고 특정 이벤트 발생에 의해 함수를 호출하고 처리하도록 하는 함수이다. 대표적으로 조기 종료를 뜻하는 Early Stopping과 학습률을 조정하는 LRScheduler가 있다. 아래에서 조기종료, 콜백 클래스를 구현한 코드를 살펴보겠다. # google colab gpu 환경 import torch import torch.nn as nn import torch.optim as optim import torchvision.models as models from torchvision import transforms, datasets import matplotlib impo..