Skip to content
This repository has been archived by the owner on Oct 13, 2021. It is now read-only.

How to cast the INT64 weights down to INT32 #726

Open
Neronjust2017 opened this issue Jun 22, 2021 · 0 comments
Open

How to cast the INT64 weights down to INT32 #726

Neronjust2017 opened this issue Jun 22, 2021 · 0 comments

Comments

@Neronjust2017
Copy link

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!

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant