fp16-convert.py 416 Bytes
Newer Older
sugon_cxj's avatar
sugon_cxj committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
import os
import onnxmltools
from onnxmltools.utils.float16_converter import convert_float_to_float16

input_onnx_model = "./model.onnx"
output_onnx_model = "./model_fp16.onnx"
print(input_onnx_model)
print(output_onnx_model)
# Load your model
onnx_model = onnxmltools.utils.load_model(input_onnx_model)

onnx_model = convert_float_to_float16(onnx_model)

onnxmltools.utils.save_model(onnx_model, output_onnx_model)