We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 90d15a6 commit f0873caCopy full SHA for f0873ca
examples/post_training_quantization/torch_fx/resnet18/main.py
@@ -194,7 +194,7 @@ def transform_fn(data_item):
194
example_input = torch.ones(*input_shape).to(device)
195
196
with disable_patching():
197
- fx_model = torch.export.export(model.eval(), args=(example_input,)).module()
+ fx_model = torch.export.export_for_training(model.eval(), args=(example_input,)).module()
198
quantized_fx_model = nncf.quantize(fx_model, quantization_dataset)
199
quantized_fx_model = torch.compile(quantized_fx_model, backend="openvino")
200
0 commit comments