pytorch查看model结构

你可以使用以下代码来查看 PyTorch 模型的结构：

import torch
import torchvision.models as models

# 加载预训练的模型
model = models.resnet18(pretrained=True)

# 打印模型结构
print(model)

这里以 ResNet-18 为例，你可以根据自己的模型替换 `models.resnet18` 部分。通过打印 `model` 可以查看到模型的结构信息，包括每一层的名称、类型和参数数量等。