Skip to content

Commit

Permalink
fix training expected int8 op
Browse files Browse the repository at this point in the history
  • Loading branch information
echarlaix committed Nov 29, 2023
1 parent cbdfe1c commit 815ede4
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/openvino/test_training.py
Original file line number Diff line number Diff line change
Expand Up @@ -361,7 +361,7 @@ def tearDown(self):
"default_quantization,structured_movement_sparsity": OVTrainerTestDescriptor(
model_id="hf-internal-testing/tiny-random-bert",
nncf_compression_config=[DEFAULT_QUANTIZATION_CONFIG, STRUCTURED_MOVEMENT_SPARSITY_CONFIG_FOR_BERT],
expected_fake_quantize=43,
expected_fake_quantize=42,
expected_int8=32,
expected_binary_masks=60,
compression_metrics=["compression_loss"],
Expand Down

0 comments on commit 815ede4

Please sign in to comment.