We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
quantize_pt2
1 parent a29dc49 commit 241cd0cCopy full SHA for 241cd0c
backends/cadence/aot/compiler.py
@@ -131,7 +131,10 @@ def quantize_pt2(
131
Prepare, convert and fuse the model using the given quantizer.
132
Returns a GraphModule with the quantized model.
133
"""
134
- # Quantizer
+ # Make the model inference mode by calling model.eval()
135
+ model.eval()
136
+
137
+ # Instantiate the quantizer to CadenceQuantizer if not supplied
138
if not quantizer:
139
quantizer = CadenceQuantizer()
140
0 commit comments