[PyTorch] 심화 문제 transfer learning을 하는데 있어서 가장 마지막 fully connected layer(이하 fc)를 변경하며, weight를 불러오는 건가? 기존 모델(이하 model)의 fc는 input 512, output 1000 이고, 변경한 모델은 input 512, output = 10이 된다.
torch.nn.init은 공식 문서를 읽어보니, input feature과 output feature을 이용, 나온 a값에 의거해 uniform 분포에서 sampling하여 weight를 채워준다고 이해했다.