Skip to content

Commit

Permalink
Merge branch 'nm/align_npu_with_cpu' of https://github.com/KodiaqQ/nncf
Browse files Browse the repository at this point in the history
… into nm/align_npu_with_cpu
  • Loading branch information
KodiaqQ committed Mar 20, 2024
2 parents c8fbfb3 + 64c5f41 commit d1c6db8
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tests/torch/quantization/test_hawq_precision_init.py
Original file line number Diff line number Diff line change
Expand Up @@ -453,8 +453,8 @@ def test_hawq_hw_npu_config_e2e(_seed, dataset_dir, tmp_path):
(
HAWQTestParams(200, 13, 100, 1.2741253547860323, 1.274125503581261),
HAWQTestParams(2, 13, 100, 1.2646427814393832, 1.2646428162034615),
HAWQTestParams(2, 10, 10, 1.830527384351921, 1.8305243724338203),
HAWQTestParams(2, 10, 5, 1.830527384351921, 1.8305243724338203),
HAWQTestParams(2, 10, 10, 1.8305239315315918, 1.8305243724338203),
HAWQTestParams(2, 10, 5, 1.8305239315315918, 1.8305243724338203),
),
ids=("until_threshold", "until_num_iter", "batch_eq_num_data", "batch_larger_num_data"),
)
Expand Down

0 comments on commit d1c6db8

Please sign in to comment.