Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add torch.set_num_threads for ptq tests (#1995)
### Changes Set number of thread for torch backends for `torch.set_num_threads(int(cpu_threads_num))` by set `CPU_THREADS_NUM` env variable. Example of reducing quantization time for TORCH backend: Model | Before | After --- | --- | --- deit3_small_patch16_224 | 2:34:09 | 0:00:46 dla34 | 0:16:38 | 0:00:38 Build: manual/post_training_quantization/122 Renamed tests to `test_ptq_quantization` ### Reason for changes Quantization time is dramatically slow for TORCH backends on CI.
- Loading branch information