• J
    Update hapi to support AMP (#31417) · fabdb43c
    Jiaqi Liu 提交于
    * make hapi support amp, and add unittest
    
    * make unittest only support GPU
    
    * update parameters for amp in hapi.Model
    
    * update hapi.Model.prepare interface, and update unittest
    
    * fix test_model.py unittest bug
    
    * add grad clear in dygraph
    
    * use_fp16_guard defaults to True, which could avoid nan
    
    * add input check, and add internal doc link to low level api
    
    * update doc, and decrease the sample num of dataset to avoid timeout
    
    * make hapi amp param  support str 'O1' or 'O2'
    
    * resume calling , modify the code of the check part
    
    * upgrade the usage of Fleet API, and disable 'pure_fp16' param
    fabdb43c
test_dist_hapi_model.py 4.0 KB