树莓派是否支持飞桨的动态图？动态图的模型如何保存被Paddle lite支持？
收藏
树莓派是否支持飞桨的动态图？动态图的模型如何保存被Paddle lite支持？
旧鸱c_
已解决
2#
回复于2021-08
支持。动态图需要通过paddle.jit.save来保存。对于动态图可参考以下例子来保存模型。 #保存静态图模型, 用于部署 x_spec = paddle.static.InputSpec(shape=[None, 3, 224, 224], name='img') # 定制化预测模型导出 model = paddle.jit.to_static(model, input_spec=[x_spec]) paddle.jit.save(model, "MyCNN")
0
收藏
请登录后评论
支持。动态图需要通过paddle.jit.save来保存。对于动态图可参考以下例子来保存模型。
#保存静态图模型, 用于部署
x_spec = paddle.static.InputSpec(shape=[None, 3, 224, 224], name='img') # 定制化预测模型导出
model = paddle.jit.to_static(model, input_spec=[x_spec])
paddle.jit.save(model, "MyCNN")