I believe you are asking how to make sure the performance of the trt engine is similar to the result of tlt model.
Please refer to Inferring resnet18 classification etlt model with python - #10 by jazeel.jk and Inferring resnet18 classification etlt model with python - #41 by Morganh