• Z
    [AMP] add state_dict and load_state_dict and unittest for class GradScaler (#34300) · 99f8f5c8
    zhangbo9674 提交于
    * add state_dict and load_state_dict and unittest for class GradScaler
    
    * refine unittest for coverage of load_state_dict
    
    * refine comments of code-block
    
    * refine some comments
    
    * refine state_dict code and unittest
    
    * add #require gpu, xpu for GradScaler get/set example code
    
    * add #require gpu, xpu for GradScaler get/set example code
    
    * refine example code
    
    * refine unittest for state_dict
    
    * refine unittest for state_dict
    
    * fix bug of DataLoader in TestGradScalerStateDict
    
    * add flag FLAGS_cudnn_deterministic
    99f8f5c8
grad_scaler.py 21.8 KB