先导包，pip install torchsummary
然后
from torchsummary import summary
model = FPN()
y = summary(model, (3, 224, 224), device="cpu")
print(y)
torchsummary使用
免责声明！
本站转载的文章为个人学习借鉴使用，本站对版权不负任何法律责任。如果侵犯了您的隐私权益，请联系本站邮箱yoyou2525@163.com删除。
先导包，pip install torchsummary
然后
from torchsummary import summary
model = FPN()
y = summary(model, (3, 224, 224), device="cpu")
print(y)
本站转载的文章为个人学习借鉴使用，本站对版权不负任何法律责任。如果侵犯了您的隐私权益，请联系本站邮箱yoyou2525@163.com删除。