diff --git a/pyinfinitensor/src/pyinfinitensor/onnx.py b/pyinfinitensor/src/pyinfinitensor/onnx.py
index 7724212e..c8f5b1a5 100644
--- 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) = (