[Pytorch] 모델에 저장된 파라미터 확인, 출력 방법

파이토치에서 파라미터 값 보는법

파이토치에서 모델에 저장된 parameter의 값들을 확인하는 방법들을 살펴보겠습니다.

 

 

파라미터만 출력 : model.parameters() 이용

layer 순서대로 파라미터를 출력하려면 선언된 모델의

parameters()를 호출하여 아래와 같이 순회해주시면 됩니다.

for param in model.parameters():
    print(param)

각 layer에 저장된 weight와 bias 값들이 차례대로 잘 출력되는 것을 볼 수 있었습니다.

 

 

layer 이름과 같이 출력 : model.named_children() 이용

만일, layer의 이름과 매칭하여 저장된 파라미터를 확인하고 싶은 경우에는

아래와 같이 named_children()을 호출하여 layer 이름과 함께 출력할 수도 있습니다.

for name, child in model.named_children():
    for param in child.parameters():
        print(name, param)

여기서는 선언한 모델 구조의 layer 이름을 각각 layer1과 layer2로 명명했었는데

해당 이름들이 파라미터와 매칭되어 잘 출력되고 있는 것을 확인할 수 있었습니다.