paddlepaddle 19 动态修改模型的最后一层

万里鹏程转瞬至

已于 2022-07-14 09:38:04 修改

阅读量4.9k

点赞数

分类专栏： paddlepaddle 文章标签：深度学习模型结构

于 2022-03-24 18:36:03 首次发布

本文链接：https://blog.youkuaiyun.com/a486259/article/details/123691959

版权

paddlepaddle 专栏收录该内容

57 篇文章 ¥59.90 ¥99.00

订阅专栏

本文介绍了在PaddlePaddle中如何遍历模型并动态修改模型的最后一层，特别是在迁移学习场景下。文章提供了遍历模型的代码示例，并强调模型最后一层必须为layer对象或Sequential对象，以适应语义分割和图像分类任务。同时，给出了修改模型最后一层的代码执行输出示例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在进行迁移学习时，通常需要修改模型的最后一层，按照需求重新进行定义。但是，迁移学习需要在众多的模型中进行对比实验，而定义最后一个layer时，需要知道layer的name，和layer输入数据的shape，也就是需要对model的layer进行遍历。

1、遍历model

通过以下代码可以遍历model

import paddle
#print('飞桨框架内置模型：', paddle.vision.models.__all__)
model=paddle.vision.resnet18()
layer_list=[(name,atom_layer) for name,atom_layer in model.named_sublayers()]
#print(model)

feature_layer_name,feature_layer=layer_list[-1]
feature_layer_shape=feature_layer.weight.shape
feature_layer_type=feature_layer.__class__
print("feature layer:",feature_layer_name,feature_layer_shape,feature_layer_type)

代码执行输出如下所示：