tensorrt_inference
tensorrt_inference copied to clipboard
ONNX conversion warning [CAN'T RECORD THE DATA FLOW OF PYTHON VALUES]
Converting a tensor to a Python boolean might cause the trace to be incorrect. We can't record the data flow of Python values, so this value will be treated as a constant in the future. This means that the trace might not generalize to other inputs! if self.grid[i].shape[2:4] != x[i].shape[2:4]:
facing this issue every time I try to convert model to onnx. Can this be ignored?
please try to use https://github.com/linghu8812/ScaledYOLOv4 to export ONNX model, the yolo.py
file is different.
Used the same[cloned the modified repository by you], but still facing the same issue. Will it affect the trt performance?
this may caused by these lines: https://github.com/linghu8812/ScaledYOLOv4/blob/b584a70b27a75e01ab5710b7743ca325f81aa443/models/yolo.py#L46-L47
@chandu1263 did it have an impect on the trt performance ?
Nope. You can ignore that actually.