torchsummary使用

先导包，pip install torchsummary
然后

from torchsummary import summary
model = FPN()
y = summary(model, (3, 224, 224), device="cpu")
print(y)