Topic
PyTorch
2 verified cases for PyTorch.
-
Fix YOLO26 ONNX opset 18 TensorRT 8.5 Concat Shape Mismatch on node_cat_25
TensorRT 8.5.2 fails building a YOLO26 ONNX opset 18 engine when Concat node_cat_25 sees [1,1,1] instead of [batch,8400,1]. Fix the export post-processing.
-
Fix RGB vs BGR and Normalization Mismatch Between Training and Inference in Computer Vision
Model looks fine in training but fails in production? Fix RGB/BGR channel order and mean/std normalization mismatches between PyTorch, OpenCV, and torchvision preprocessing.