先导包,pip install torchsummary 然后
pip install torchsummary
from torchsummary import summary model = FPN() y = summary(model, (3, 224, 224), device="cpu") print(y)