Closed YirongMao closed 4 years ago
Hi, I serialized the model, and then loaded the model in C++ API following https://github.com/NVIDIA-AI-IOT/torch2trt/issues/16
But when I sent a tensor with all ones into the models (pytorch model and serialized model in c++), their outputs are different. when I sent a tensor with all zeros, the outputs are the same.
My TensorRT version is 5.0.2.6 PyTorch version is 1.1.0
So sorry. The outputs are the same. There is a problem in my code.
Hi, I serialized the model, and then loaded the model in C++ API following https://github.com/NVIDIA-AI-IOT/torch2trt/issues/16
But when I sent a tensor with all ones into the models (pytorch model and serialized model in c++), their outputs are different. when I sent a tensor with all zeros, the outputs are the same.
My TensorRT version is 5.0.2.6 PyTorch version is 1.1.0