keras-onnx icon indicating copy to clipboard operation
keras-onnx copied to clipboard

How to cast the INT64 weights down to INT32

Open Neronjust2017 opened this issue 3 years ago • 0 comments

Hi, I convert my keras model to onnx using keras2onnx. However, when I try to convert the onnx file to TensorRT file, I encountered the following warnning "Your ONNX model has been generated with INT64 weights, while TensorRT does not natively support INT64. Attempting to cast down to INT32". How can I get a onnx model converted from keras with INT32 weights? Thanks!

Neronjust2017 avatar Jun 22 '21 09:06 Neronjust2017