首页 Paddle框架 帖子详情
如何在GPU训练的模型后用保存,然后用CPU版导入
收藏
快速回复
Paddle框架 问答模型训练 303 2
如何在GPU训练的模型后用保存,然后用CPU版导入
收藏
快速回复
Paddle框架 问答模型训练 303 2

如何在GPU训练的模型后用保存,然后用CPU版导入,像Pytorch一样,通过该设备就行?

0
收藏
回复
全部评论(2)
时间顺序
Bohiizin
#5 回复于2022-05

用***.state_dict()和***.set_state_dict()两个API。

例如:

from paddle import nn

from paddle import save,load

net=nn.Linear(3,3)

 

save(net.state_dict(),'lorem_ipsum.dolor')                 #先用net.state_dict()获取网络的参数列表,再将网络参数存储在文件'lorem_ipsum.dolor'中

 

sd=load('lorem_ipsum.dolor')            #读取存储在文件'lorem_ipsum.dolor'中的参数列表

net.set_state_dict(sd)                           #将参数列表加载到网络中,这个API在pytorch中写作net.load_state_dict(sd)

0
回复
T
Tedoliy
#7 回复于2022-09

和pytorch几乎一样的

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户