onnx / keras-onnx

Convert tf.keras/Keras models to ONNX
Apache License 2.0
379 stars 110 forks source link

How to cast the INT64 weights down to INT32 #726

Open Neronjust2017 opened 3 years ago

Neronjust2017 commented 3 years ago

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!