From 886c5e5bd2b7a8b3fadec576fdc4288efa84ddc7 Mon Sep 17 00:00:00 2001
From: Haojie Wang <haojie0429@gmail.com>
Date: Tue, 16 Jan 2024 06:33:03 +0000
Subject: [PATCH] add hack comment

---
 pyinfinitensor/src/pyinfinitensor/onnx.py | 2 ++
 1 file changed, 2 insertions(+)

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) = (