pytorch如何加载自己训练的模型

在resnet上添加了注意力机制然后训练得到了一个.pth格式的模型，我想请问下，如何加载这个模型并打印这个模型的权重。
如，
import torchvision.models as models

model = models.vgg16(pretrained = True)

print(model)
这个可以输出vgg16的模型权重层。