• W
    Add API for quantization-aware training in dygraph mode (#49398) · b53888e7
    whs 提交于
    * Add tools for quantization-aware training
    1. Expose an API named paddle.quantization.QAT
    2. Define a wrapper class to insert quanters into model for QAT
    3. Add some functions in QuantConfig for QAT
    4. Add unittest for QAT
    
    * Add QuantedConv2D and QuantedLinear for QAT
    
    * Add paddle.nn.quant.qat to setup.py
    b53888e7
wrapper.py 1.7 KB