Add DNN warning comment (ultralytics#10368)

Update export.py Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
sanweiw · Dec 1, 2022 · 185d475 · 185d475
1 parent 028b7cd
commit 185d475
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/export.py b/export.py
@@ -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)