paddle如何同时加载多个自己的预训练模型?
军人的生活就是 发布于2021-01 浏览:419 回复:4
0
收藏

假设我现在有三个预训练模型A,B,C。然后有三个独立的文件分别对应这三个模型预训练的参数。现在我想在自己的神经网络中同时使用这三个预训练模型,要怎么样进行加载呢?

modelA = PretainModelA()
modelB = PretainModelB()
modelC = PretainModelC()

paddle.fluid.io.load_vars(exe, model_path_A, main_progaram) 

上面的伪代码,是只加载了模型A么?如果要分别加载A、B、C三个模型该怎么处理呢?

收藏
点赞
0
个赞
共4条回复 最后由189******30回复于2021-01
#6189******30回复于2021-01

静态图没怎么用过(哭脸)

0
#4189******30回复于2021-01

可以参考文档

https://www.paddlepaddle.org.cn/documentation/docs/zh/develop/guides/02_paddle2.0_develop/08_model_save_load_cn.html

0
#3189******30回复于2021-01

2.0动态图版本就声明三个模型对象,然后分别加载就可以了,非常直观

0
#2189******30回复于2021-01

用2.0版本吧,以后fluid下的api可能很快不支持了

0
TOP
切换版块