tests/runner/test_runner_default_runner.py · ef9c20cc18dbaa100ca8f7a4057a95324c7abc66 · OpenDAS / d2go

Peizhao Zhang authored Oct 20, 2021

Summary:
Refactored qat related code.
* Moved `_prepare_model_for_qat` related code to a function.
* Moved `_setup_non_qat_to_qat_state_dict_map` related code to a function.
* Moved QATHook related code to the quantization file and implemented as a class.

Differential Revision: D31370819

fbshipit-source-id: 836550b2c8d68cd93a84d5877ad9cef6f0f0eb39

ef9c20cc

test_runner_default_runner.py 14.2 KB

Replace test_runner_default_runner.py