在resnet上添加了注意力机制然后训练得到了一个.pth格式的模型,我想请问下,如何加载这个模型并打印这个模型的权重。
如,
import torchvision.models as models
model = models.vgg16(pretrained = True)
print(model)
这个可以输出vgg16的模型权重层。
在resnet上添加了注意力机制然后训练得到了一个.pth格式的模型,我想请问下,如何加载这个模型并打印这个模型的权重。
如,
import torchvision.models as models
model = models.vgg16(pretrained = True)
print(model)
这个可以输出vgg16的模型权重层。