[TensorRT] Your ONNX model has been generated with INT64 weights, while TensorRT does not natively support INT64. Attempting to cast down to INT32.
|2022. 2. 23. 20:39
728x90
반응형
ONNX를 TRT로 변환하면 아래와 같은 Warning 메세지가 뜬다.
Your ONNX model has been generated with INT64 weights,
while TensorRT does not natively support INT64. Attempting to cast down to INT32.
성능상의 손실은 없는 것으로 판단되나 정말 성능에 영향이 없는지 실험이 필요하다. 왜 뜨는 것일까?
관련 이슈 : https://github.com/onnx/tensorflow-onnx/issues/883#issuecomment-614561227
onnx-simplifier를 사용하여 ONNX 모델을 심플하게 변환해보는 것도 방법일 듯 하다.
https://github.com/NVIDIA/TensorRT/issues/284
https://github.com/NVIDIA/TensorRT/issues/284#issuecomment-572835659
728x90
반응형
'AI Development > TensorRT' 카테고리의 다른 글
[TensorRT] trtexec dumpProfile (0) | 2022.03.22 |
---|---|
[NVIDIA TAO Toolkit] TAO(Train, Adapt, and Optimize) Toolkit (0) | 2022.03.15 |
[TensorRT] ONNX 및 TRT에서 Group Normalization 사용하기 (+ Instance Normalization 이슈) (0) | 2022.02.23 |
[TensorRT] QuickStartGuide (6) | 2021.08.31 |
[TensorRT] trtexec 사용하기 (2) | 2021.03.30 |