• Y
    [AMP] Allow to enable multi_precision through paddle.static.amp.decorate and... · 680460fd
    Yiqun Liu 提交于
    [AMP] Allow to enable multi_precision through paddle.static.amp.decorate and add documents for some apis. (#53012)
    
    * Add document for some apis. test=docs_preview
    
    * Allow to set master_weight in paddle.static.amp.decorate.
    
    * Polish codes and add unittest.
    
    * Refine docs.
    
    * Remove the repetitive function.
    680460fd
test_model_cast_to_bf16.py 10.3 KB