先导包，pip install torchsummary 然后
pip install torchsummary
from torchsummary import summary model = FPN() y = summary(model, (3, 224, 224), device="cpu") print(y)
torchsummary使用
原文：https://www.cnblogs.com/consolexinhun/p/14258869.html
