fine tuning이 아닐 경우 전체 수행 epoch횟수의 의미

안녕하세요, 인프런 AI 인턴입니다.
shin643님의 질문에 답변드리겠습니다.

Config 클래스의 N_EPOCHS는 모델이 학습하는 총 epoch 수를 의미합니다. 여기서 fine tuning이 아닌 경우, N_EPOCHS = 30은 이미지넷(ImageNet) 등의 사전 학습된(pretrained) 모델을 사용하지 않고, 모델의 가중치(weight parameters)를 처음부터 끝까지(즉, scratch으로) 학습할 때 총 30번의 epoch을 수행한다는 것을 의미합니다.

즉, fine tuning의 경우 사전에 학습된 모델을 기반으로 추가 학습을 하는 것이고, 여기서 언급된 N_EPOCHS = 30은 사전 학습 없이 모델을 처음부터 학습시킬 때, 모델이 전체 데이터셋을 몇 번 학습하는지에 대한 횟수를 설정하는 것입니다.

추가적으로, First_EPOCHS와 Second_EPOCHS는 fine tuning 과정에서 사용되며, 각각 첫 번째 단계와 두 번째 단계에서 모델이 학습하는 epoch 수를 의미합니다. Fine tuning 과정에서 보통 두 단계로 나누어 학습하는데, 이때 각 단계별로 학습을 얼마나 진행할지 epoch 수를 정의하는 것입니다.

좋은 질문 감사합니다. 더 궁금하신 사항이 있으면 언제든지 문의 주세요.