TensorRT convert and Test

[BlackLee68]

hi @wenbowen123 , thanks for the great work!

I tested the model on AGX orin 32G(Jetpack 6.2), I found the 20-30-48 and the 23-36-67 shows different results when using TRT model for testing, I use the same convert code and command to generate TRT engine, but the final results seemed quite different when using run_demo_tensorrt.py, below is the res:

1, 23-36-67 Checkpoint with 4 iters, the result looks good:

2, 20-30-48 Checkpoint with 4 iters, the result looks a little strange:

I checked the post_runner, found the input is different between 23-36-67 and 20-30-48, the "features_left_16" is not used in 20-30-48 TRT engine, is this normal or i missed something when converting the model ?

Thanks a lot！

[te-shi]

Hi, I encountered the same issue. From my testing, if you are using the 20-26-39 or 20-30-48 checkpoints, you need to set normalize=False in the build_gwc_volume_triton function. Hope this helps.

[wenbowen123]

Hi, this should have been fixed in latest update.

[BlackLee68]

thanks, that helps. @wenbowen123 any plan to release training code?

[runner-9]

Hello, may I ask if your jetpack is 6.2.2? What version of tensorrt is there? The model operation result I obtained by converting with origin jetpack6.2.2 and tensorrt 10.3 is completely abnormal.