Skip to content
Snippets Groups Projects
Commit 078c8ef0 authored by auphelia's avatar auphelia
Browse files

[CustomOp] Delte rounding in execute node of QuantAvgPool2d

parent 549a98cf
No related branches found
No related tags found
No related merge requests found
......@@ -66,7 +66,7 @@ class QuantAvgPool2d(CustomOp):
outputs=[outp],
)
model_avgpool = helper.make_model(graph_avgpool)
idict = {node.input[0]: np.round(context[node.input[0]])}
idict = {node.input[0]: context[node.input[0]]}
sess = rt.InferenceSession(model_avgpool.SerializeToString())
result_temp = sess.run(None, idict)
# remove scaling introduced by average
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment