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

add hack comment

This commit is contained in:
Haojie Wang 2024-01-16 06:33:03 +00:00
parent 9b63a62b70
commit 886c5e5bd2
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -952,6 +952,8 @@ class OnnxStub:
size, size,
) )
elif node.op_type == "Dropout": elif node.op_type == "Dropout":
// HACK: Currently we only support model inference,
// so training_mode is set to false.
attributes = _parse_attribute( attributes = _parse_attribute(
node, {"ratio": 0.5, "training_mode": 0}) node, {"ratio": 0.5, "training_mode": 0})
(ratio, training_mode) = ( (ratio, training_mode) = (