假设我现在有三个预训练模型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
个赞
请登录后评论
TOP
切换版块
静态图没怎么用过(哭脸)
可以参考文档
https://www.paddlepaddle.org.cn/documentation/docs/zh/develop/guides/02_paddle2.0_develop/08_model_save_load_cn.html
2.0动态图版本就声明三个模型对象,然后分别加载就可以了,非常直观
用2.0版本吧,以后fluid下的api可能很快不支持了