Fix missing tensor_input (#14790)

This commit is contained in:
Nicolas Mowen authored and GitHub committed 2024-11-04 13:04:33 -06:00
1 parent a13b9815f6
commit 553676aade
1 file changed
+1
@@ -457,6 +457,7 @@ model:
width: 320 # <--- should match whatever was set in notebook
height: 320 # <--- should match whatever was set in notebook
input_pixel_format: bgr
input_tensor: nchw
path: /config/yolo_nas_s.onnx
labelmap_path: /labelmap/coco-80.txt
```