torchsummary安装包

您可以使用以下命令安装torchsummary包：

pip install torchsummary

安装完成后，您可以在代码中使用以下语句导入：

from torchsummary import summary

然后，您可以使用`summary`函数来打印模型的摘要信息，例如：

import torch.nn as nn
import torchvision.models as models

model = models.resnet18()
summary(model, (3, 224, 224))

输出结果将显示模型的层次结构、每一层的输出形状以及总参数数量等信息。