Skip to content

Commit

Permalink
test fix
Browse files Browse the repository at this point in the history
  • Loading branch information
bfineran committed Jul 27, 2023
1 parent 1880f48 commit 0b327f1
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/sparseml/pytorch/utils/exporter.py
Original file line number Diff line number Diff line change
Expand Up @@ -581,12 +581,12 @@ def export_onnx(

if convert_qat and is_quant_module:

use_qlinearconv = hasattr(module, "export_with_qlinearconv") and (
use_qlinear_conv = hasattr(module, "export_with_qlinearconv") and (
module.export_with_qlinearconv
)

exporter = ONNXToDeepsparse(
use_qlinearconv=use_qlinearconv,
use_qlinear_conv=use_qlinear_conv,
skip_input_quantize=skip_input_quantize,
)
exporter.export(pre_transforms_model=file_path, file_path=file_path)
Expand Down

0 comments on commit 0b327f1

Please sign in to comment.