先导包，pip install torchsummary
然后
from torchsummary import summary
model = FPN()
y = summary(model, (3, 224, 224), device="cpu")
print(y)
相关文章：
相似解决方案
热门标签
先导包，pip install torchsummary
然后
from torchsummary import summary
model = FPN()
y = summary(model, (3, 224, 224), device="cpu")
print(y)
相关文章：