[pytorch] summary 'list' object has no attribute ' size' error

2021. 8. 18. 18:26

pytorch에서 keras처럼 summary를 확인하는 함수가 있다. 이때 가장 많이 사용하는 함수가 아래의 함수일 것이다.

from torchsummary import summary

 


필자도 torchsummary를 사용하여 model의 summary를 확인했다. 그러나 Hrnet을 설계할 때 값이 리스트 형태로 전달하면서 다음과 같은 오류가 발생했다.

File "\torchsummary\torchsummary.py", line 19, in hook
    summary[m_key]["input_shape"] = list(input[0].size())
AttributeError: 'list' object has no attribute 'size'

그래서 구글에서 검색 중 github에서 해결 방법을 찾았다. 해결방법은 의외로 간단하다. torchsummaryX를 사용하면 된다.  아래 코드와 같이 이용하면 리스트 형태로 전달해도 Summary를 확인 할 수 있다.

from torchsummaryX import summary
arch = summary(model, torch.rand((1, 3, 256, 256)))

 


참고문헌

 

 

AttributeError: 'list' object has no attribute 'size' · Issue #137 · qubvel/segmentation_models.pytorch

i used the same code below,just replaced se_resnext50_32x4d with vgg11/vgg16 and i can get the model summary for unet with vgg11/vgg16 but whenever i try to get model summary of FPN with se_resnext...

github.com