This repository was archived by the owner on Oct 13, 2021. It is now read-only.
This repository was archived by the owner on Oct 13, 2021. It is now read-only.
How to cast the INT64 weights down to INT32 #726
Description
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!