paddle静态图模型的保存和加载

##静态图模型的保存与加载

"""
在静态图中，模型结构部分可以转化为可持久化的程序
"""

import paddle
import paddle.static as static

#开启静态图模式
paddle.enable_static()

#创建输入数据和网络
x=paddle.static.data(name='x',shape=[None,224],dtype='float32')
z=paddle.static.nn.fc(x,10)

print(x)
print(z)
#设置执行器开始训练
place=paddle.CPUPlace()
exe=paddle.static.Executor(place)
print(exe)
exe.run(paddle.static.default_startup_program())
prog=paddle.static.default_main_program()

##保存模型的参数
paddle.save(prog.state_dict(),'temp/model.pdparms')
print('静态图模型参数',prog.state_dict(),len(prog.state_dict()))
#载入模型参数
state_dict=paddle.load('temp/model.pdparms')
prog.set_state_dict(state_dict)
print('导入的模型参数',state_dict)

##保存整个静态图模型（包含静态图结构和参数）
paddle.save(prog,'temp/model.pdmodel')
print('整个静态图模型',prog)
#导入整个模型结构
prog=paddle.load('temp/model.pdmodel')
print('导入的模型结构',prog)