[딥러닝] Pytorch 모델 구조 및 파라미터 확인
파이토치(Pytorch)를 이용한 모델 학습 시 사용되는 모델의 구조, 파라미터(parameters)의 이름 및 개수를 확인하고 싶을 때 아래의 파이썬 코드를 통해 확인 가능하다. 방법 1 for parameter in model.parameters(): print(parameter) 방법 2 from torchsummary import summary summary(model, (1,)) # inputs: model, input_size=(1,) summary(model, (3,224,224)) # inputs: model, input_size=(3,224,224) 간단한 polynomial의 경우 입력의 차원은 (1,) 224x224 RGB 이미지의 경우 입력의 차원은 (3, 224, 224)
2022. 6. 25.
