可变data_type
概述
可变data_type是OMG工具支持的一个功能,用于模型输入输出数据类型多样性的场景,无需修改训练好的模型,在使用OMG工具进行模型转换时,通过指定输入、输出数据类型使得同一个模型适用于不同输入输出的场景。
使用说明
在进行模型转换时,输入输出数据类型指定分别需要通过OMG参数的input_type、output_type来实现。
使用示例:
./omg --model=./model.pb --framework=3 --output=./model --input_shape="inputs:1,512,512,1" --out_nodes="outputs:0" --input_type="inputs:FP16" --output_type="outputs:UINT8"