PyTorch

사용할 데이터셋은 개와 고양이로 이루어져 있으며 여기서 받을 수 있습니다. Pretrained 모델을 불러오고 설정하기 그저 가중치를 불러올 뿐만 아니라, 분류 모델에서 일부 레이어들을 고정하고 분류할 클래스 수에 맞추어 말단 레이어를 수정하는 과정이 필요합니다.

모델을 학습하기 전에 모델의 가중치를 특정 분포로 초기화를 해주면, 학습 속도나 정확도 등 학습에 도움이 될 수 있습니다. 여기서는 모델의 가중치를 초기화하는 방법에 대해 알아보겠습니다. 먼저 텐서를 특정 분포로 초기화하는 함수들을 알아보겠습니다.