Skip to content

Commit 8fb8cfe

Browse files
committed
fix quant op
1 parent 8911601 commit 8fb8cfe

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/openvino/test_training.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -730,7 +730,7 @@ def check_ovmodel_reshaping(self, ovmodel: OVModel):
730730
"quantization": OVTrainerTestDescriptor(
731731
model_id="hf-internal-testing/tiny-random-Wav2Vec2Model",
732732
nncf_compression_config=[QUANTIZATION_CONFIG_FOR_WAV2VEC2],
733-
expected_fake_quantize=40,
733+
expected_fake_quantize=24,
734734
expected_int8=30,
735735
compression_metrics=["compression_loss"],
736736
),
@@ -757,7 +757,7 @@ def check_ovmodel_reshaping(self, ovmodel: OVModel):
757757
"quantization,unstructured_movement_sparsity": OVTrainerTestDescriptor(
758758
model_id="hf-internal-testing/tiny-random-Wav2Vec2Model",
759759
nncf_compression_config=[QUANTIZATION_CONFIG_FOR_WAV2VEC2, UNSTRUCTURED_MOVEMENT_SPARSITY_CONFIG_FOR_WAV2VEC2],
760-
expected_fake_quantize=40,
760+
expected_fake_quantize=24,
761761
expected_int8=30,
762762
expected_binary_masks=48,
763763
compression_metrics=["compression_loss"],
@@ -775,7 +775,7 @@ def check_ovmodel_reshaping(self, ovmodel: OVModel):
775775
model_id="hf-internal-testing/tiny-random-Wav2Vec2Model",
776776
teacher_model_id="hf-internal-testing/tiny-random-Wav2Vec2Model",
777777
nncf_compression_config=[QUANTIZATION_CONFIG_FOR_WAV2VEC2, UNSTRUCTURED_MOVEMENT_SPARSITY_CONFIG_FOR_WAV2VEC2],
778-
expected_fake_quantize=40,
778+
expected_fake_quantize=24,
779779
expected_int8=30,
780780
expected_binary_masks=48,
781781
compression_metrics=["compression_loss", "distillation_loss", "task_loss"],

0 commit comments

Comments
 (0)