[pytorch] 내가 보려고 정리하는 파이토치 - torchvisions.models 내부 weight 보는 법

내가 보려고 정리하는 파이토치 00 - torchvision.models 에서 내부 weight 보는 법 

import torchvision.models as models

이렇게 import 하면 pretrained 된 모델 zoo를 이용할 수 있다

self.backbone = models.vgg16(pretrained=True, progress=True)

근데 이런식으로 불러온 다음에 self.backbone을 출력하면 weight를 볼 수 있는게 아니라 

이렇게 모델 구성 dictionary를 보여준다. 

.items()로 내부 parameter를 확인할 수 있는 줄 알았는데 아니였고 ㅜ 

 

해결방법은 아래와 같이 내부 값을 .data로 옮겨서 확인할 수 있다. 

 

for w in backbone.parameters():
    w = w.data.cpu()

 

결과는 요런식으로 

 

 

끝 ~