p35189024
/
InfiniTensor
forked from jiuyuan/InfiniTensor
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix comment

This commit is contained in:
Haojie Wang 2024-01-17 02:57:44 +00:00
parent 886c5e5bd2
commit e33131ce5c
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
pyinfinitensor/src/pyinfinitensor/onnx.py
View File

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