标签：lan import rom int code pre src device loading
先导包，pip install torchsummary
然后
from torchsummary import summary
model = FPN()
y = summary(model, (3, 224, 224), device="cpu")
print(y)
标签：lan import rom int code pre src device loading
原文地址：https://www.cnblogs.com/consolexinhun/p/14258869.html
