pytorch 打印模型结构

在PyTorch中，可以通过打印模型对象的字符串表示来查看模型的结构。这可以通过调用模型对象的`__str__()`或`__repr__()`方法来实现。

例如，假设我们有一个名为`model`的PyTorch模型对象，我们可以通过以下方式打印模型结构：

print(model)

这将打印模型的字符串表示，其中包含模型的架构和每个层的参数数量。另外，可以使用`torchsummary`库来更详细地打印模型结构和参数数量。例如：

from torchsummary import summary

summary(model, input_size=(3, 224, 224))

这将打印出模型的详细结构，包括每个层的输入和输出形状，以及模型的总参数数量和内存使用情况。