add hack comment

2024-01-16 06:33:03 +00:00 · 2024-01-16 06:33:03 +00:00 · 886c5e5bd2
parent 9b63a62b70
commit 886c5e5bd2
1 changed files with 2 additions and 0 deletions
--- a/pyinfinitensor/src/pyinfinitensor/onnx.py
+++ b/pyinfinitensor/src/pyinfinitensor/onnx.py
@ -952,6 +952,8 @@ class OnnxStub:
                    size,
                )
            elif node.op_type == "Dropout":
+                // HACK: Currently we only support model inference,
+                // so training_mode is set to false.
                attributes = _parse_attribute(
                        node, {"ratio": 0.5, "training_mode": 0})
                (ratio, training_mode) = (