@@ -365,7 +365,7 @@ def tearDown(self):
365
365
"default_quantization,structured_movement_sparsity" : OVTrainerTestDescriptor (
366
366
model_id = "hf-internal-testing/tiny-random-bert" ,
367
367
nncf_compression_config = [DEFAULT_QUANTIZATION_CONFIG , STRUCTURED_MOVEMENT_SPARSITY_CONFIG_FOR_BERT ],
368
- expected_fake_quantize = 44 ,
368
+ expected_fake_quantize = 34 ,
369
369
expected_int8 = 32 ,
370
370
expected_binary_masks = 60 ,
371
371
compression_metrics = ["compression_loss" ],
@@ -376,7 +376,7 @@ def tearDown(self):
376
376
CUSTOMIZED_QUANTIZATION_CONFIG ,
377
377
STRUCTURED_MOVEMENT_SPARSITY_CONFIG_FOR_BERT ,
378
378
],
379
- expected_fake_quantize = 44 ,
379
+ expected_fake_quantize = 34 ,
380
380
expected_int8 = 32 ,
381
381
expected_binary_masks = 60 ,
382
382
compression_metrics = ["compression_loss" ],
@@ -385,7 +385,7 @@ def tearDown(self):
385
385
model_id = "hf-internal-testing/tiny-random-bert" ,
386
386
teacher_model_id = "hf-internal-testing/tiny-random-bert" ,
387
387
nncf_compression_config = [DEFAULT_QUANTIZATION_CONFIG , STRUCTURED_MOVEMENT_SPARSITY_CONFIG_FOR_BERT ],
388
- expected_fake_quantize = 44 ,
388
+ expected_fake_quantize = 34 ,
389
389
expected_int8 = 32 ,
390
390
expected_binary_masks = 60 ,
391
391
compression_metrics = ["compression_loss" , "distillation_loss" , "task_loss" ],
@@ -397,7 +397,7 @@ def tearDown(self):
397
397
CUSTOMIZED_QUANTIZATION_CONFIG ,
398
398
STRUCTURED_MOVEMENT_SPARSITY_CONFIG_FOR_BERT ,
399
399
],
400
- expected_fake_quantize = 44 ,
400
+ expected_fake_quantize = 34 ,
401
401
expected_int8 = 32 ,
402
402
expected_binary_masks = 60 ,
403
403
compression_metrics = ["compression_loss" , "distillation_loss" , "task_loss" ],
@@ -574,7 +574,7 @@ def check_ovmodel_reshaping(self, ovmodel: OVModel):
574
574
nncf_compression_config = [STRUCTURED_MOVEMENT_SPARSITY_CONFIG_FOR_SWIN , DEFAULT_QUANTIZATION_CONFIG ],
575
575
expected_fake_quantize = 28 ,
576
576
expected_int8 = 28 ,
577
- expected_binary_masks = 48 ,
577
+ expected_binary_masks = 40 ,
578
578
compression_metrics = ["compression_loss" ],
579
579
),
580
580
"default_quantization,unstructured_movement_sparsity" : OVTrainerTestDescriptor (
@@ -591,7 +591,7 @@ def check_ovmodel_reshaping(self, ovmodel: OVModel):
591
591
nncf_compression_config = [STRUCTURED_MOVEMENT_SPARSITY_CONFIG_FOR_SWIN , DEFAULT_QUANTIZATION_CONFIG ],
592
592
expected_fake_quantize = 28 ,
593
593
expected_int8 = 28 ,
594
- expected_binary_masks = 48 ,
594
+ expected_binary_masks = 40 ,
595
595
compression_metrics = ["compression_loss" , "distillation_loss" , "task_loss" ],
596
596
),
597
597
"distillation,default_quantization,unstructured_movement_sparsity" : OVTrainerTestDescriptor (
0 commit comments