Add DNN warning comment (#10368)

Update export.py

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Glenn Jocher authored and GitHub committed 2022-12-01 13:01:46 -08:00
1 parent 028b7cdb5a
commit 185d475d93
1 file changed
+1 -1
+1 -1
View File
@@ -153,7 +153,7 @@ def export_onnx(model, im, file, opset, dynamic, simplify, prefix=colorstr('ONNX
f,
verbose=False,
opset_version=opset,
do_constant_folding=True,
do_constant_folding=True, # WARNING: DNN inference with torch>=1.12 may require do_constant_folding=False
input_names=['images'],
output_names=output_names,
dynamic_axes=dynamic or None)