[Pytorch] 모델 훈련 중 특정 레이어 프리징 방법 How to freeze a specific layer in a neural model at training-time?


[Pytorch] 모델 훈련 중 특정 레이어 프리징 방법 How to freeze a specific layer in a neural model at training-time?

안녕하세요. 이번 포스팅은 신경망 모델 훈련 중에 특정 레이어를 프리징하는 방법에 대해 알려드리고자 합니다. 프리징(Freezing)은 뭘까요?영어 단어 그 자체의 의미는 "얼리다"이지만 신경망 모델, 딥러닝에서는 모델이나 모듈의 파라미터를 모델 트레이닝 동안 고정하는 것을 의미합니다. 즉, 모델의 train step을 진행해도 프로그래머가 특정한 모델이나 모듈의 파라미터(weight or bias or both)는 변하지 않는 것입니다. 왜? 프리징이 필요할까?다양한 이유가 있을 수 있지만, 훈련이 완료된 모델(A) 위에 새로운 모듈(B)을 올릴 때 훈련이 완료된 모델의 파라미터(A's parameter)를 그대로 유지해야하거나 하고..........

[Pytorch] 모델 훈련 중 특정 레이어 프리징 방법 How to freeze a specific layer in a neural model at training-time?에 대한 요약내용입니다.

자세한 내용은 아래에 원문링크를 확인해주시기 바랍니다.



원문링크 : [Pytorch] 모델 훈련 중 특정 레이어 프리징 방법 How to freeze a specific layer in a neural model at training-time?