缺少“torchsummary”模块

如果你运行 PyTorch 程序时遇到了 “没有名为 <> 的模块” 的错误信息，这意味着你没有正确安装该模块。

torchsummary 是一个开源 Python 库，它可以方便地向你展示 PyTorch 模型的详细信息。它的主要功能是打印模型中每一层的形状、参数数量和输出数量等信息，以便更好地理解模型，并进行优化和调试。

为了解决该错误，你可以使用以下命令安装 torchsummary 库:

!pip install torchsummary

然后，你可以按照以下方式在你的代码中引入 torchsummary：

from torchsummary import summary

接下来，你可以使用 summary 函数来打印模型的详细信息：

summary(model, input_size=(channels, height, width))

注意，这里的 model 是你的 PyTorch 模型，而 input_size 是你的输入张量的形状。

使用 torchsummary 可以帮助你更快地了解你的模型，并找出在哪些地方可以进行优化。希望这篇文章对你有所帮助！