Unable to compile on drp-ai-tvm compilation environment the Pre-trained FasterViT based ImageNet Classification onnx weights from NVIDIA TAO-Pre-trained FasterViT based ImageNet Classification weights | NVIDIA NGC
and getting this error: Run TVM frotend compiler
compile_onnx_model_quant.py:169: DeprecationWarning: mapping.TENSOR_TYPE_TO_NP_TYPE
is now deprecated and will be removed in a future release.To silence this warning, please use helper.tensor_dtype_to_np_dtype
instead.
mod, params = relay.frontend.from_onnx(onnx_model, shape_dict)
The Relay type checker is unable to show the following types match:
Tensor[(1, 1, 3, 3), float32]
Tensor[(1, 64, 3, 3), float32]
In particular:
dimension 1 conflicts: 1 does not match 64.
The Relay type checker is unable to show the following types match.
In particular Tensor[(1, 64, 3, 3), float32]
does not match Tensor[(1, 1, 3, 3), float32]
note: run with TVM_BACKTRACE=1
environment variable to display a backtrace.