Skip to content

Fluid版本训练好的模型如何进行合并? #10990

@yeyupiaoling

Description

@yeyupiaoling

在Android设备或者是Linux下,使用的预测模型是合并的模型,在V2版本中,预测模型是一个.tar.gz的文件。但是在fluid版本的预测模型不是这样的了。那么如何合并模型呢?

以下是V2的合并模型的代码:

from paddle.utils.merge_model import merge_v2_model

# import your network configuration
from mobilenet import mobile_net

net = mobile_net(3*224*224, 102, 1.0)
param_file = './mobilenet_flowers102.tar.gz'
output_file = './output.paddle'

merge_v2_model(net, param_file, output_file)

Metadata

Metadata

Assignees

Labels

User用于标记用户问题

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions