当前位置:网站首页>paddlepaddle 19 动态修改模型的最后一层
paddlepaddle 19 动态修改模型的最后一层
2022-06-27 01:58:00 【万里鹏程转瞬至】
在进行迁移学习时,通常需要修改模型的最后一层,按照需求重新进行定义。但是,迁移学习需要在众多的模型中进行对比实验,而定义最后一个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)代码执行输出如下所示:
feature layer: fc [512, 1000] <class 'paddle.nn.layer.common.Linear'>
2、动态修改模型的最后一层
动态修改模型的最后一层,支持语义分割模型和图像分类模型。但是需要注意的是,模型的最后一层必须是layer对象(如resnet系列,densenet系列,mobilenet系
边栏推荐
猜你喜欢
随机推荐
Shell脚本系列篇(1) 入门
UVM in UVM_ report_ Enabled usage
get_ Usage Summary of sequencer
速看!2022年6月编程语言排行榜出炉!第一名太牛啦
达梦数据库安装
别被洗脑了,这才是90%中国人的工资真相
I encountered some problems when connecting to the database. How can I solve them?
两个页面之间传参方法
Oracle/PLSQL: Length Function
three.js多米诺骨牌js特效
图论知识及其应用初步调研
SystemVerilog simulation speed increase
Hot discussion: what are you doing for a meaningless job with a monthly salary of 18000?
Oracle/PLSQL: To_ Clob Function
Oracle/PLSQL: CharToRowid Function
Dameng database installation
p5.js死亡星球
Oracle/PLSQL: Replace Function
P5.js death planet
简单学习GoogleColab的入门级概念









