TensorRT YOLO inference error

@AastaLLL please take a look
https://devtalk.nvidia.com/default/topic/1037572/jetson-tx2/inference-error-using-yolov2-on-jetson-tx2/