欢迎您访问 最编程 本站为您分享编程语言代码,编程技术文章!
您现在的位置是: 首页

理解模型VGG16的feature[:-1]部分的实际含义

最编程 2024-07-23 18:43:48
...

用于记录:

pretrained_model = models.vgg16(True).features[:-1]
print(pretrained_model)

在这里插入图片描述
就是把vgg16最后一层不要了

pretrained_model = models.vgg16(True).features
print(pretrained_model)

在这里插入图片描述
torch保存的模型和参数按照字典方式保存。

推荐阅读