From e33131ce5c6f21623e23eccc68f979832ebcd104 Mon Sep 17 00:00:00 2001 From: Haojie Wang Date: Wed, 17 Jan 2024 02:57:44 +0000 Subject: [PATCH] fix comment --- pyinfinitensor/src/pyinfinitensor/onnx.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pyinfinitensor/src/pyinfinitensor/onnx.py b/pyinfinitensor/src/pyinfinitensor/onnx.py index c8f5b1a5..c5c5638c 100644 --- a/pyinfinitensor/src/pyinfinitensor/onnx.py +++ b/pyinfinitensor/src/pyinfinitensor/onnx.py @@ -952,8 +952,8 @@ class OnnxStub: size, ) elif node.op_type == "Dropout": - // HACK: Currently we only support model inference, - // so training_mode is set to false. + # 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) = (