search

NVIDIA / trt-samples-for-hackathon-cn

Simple samples for TensorRT programming
Apache License 2.0
1.47k stars 337 forks source link

failed reading calibration cache via c++ api #58

Open KexianShen opened 1 year ago

KexianShen commented 1 year ago

https://github.com/NVIDIA/trt-samples-for-hackathon-cn/blob/88fe8012cd8cc5c09a63b18ecb91be5bd89a6a9f/cookbook/04-Parser/pyTorch-ONNX-TensorRT/C%2B%2B/calibrator.cpp#L58-L73

operator >> in line 70 only reads the first word before white space or \n