• 0
    [Dy2stat]support pure fp16 for dy2stat (#36944) · 52edad6a
    0x45f 提交于
    * run dy2stat pure fp16 in Linear model
    
    * no use self._pure_fp16_inputs
    
    * add test and fix Adam error in dy2stat pure fp16 training
    
    * use paddle.optimizer.Adam
    
    * run test in gpu
    
    * change test time for CI
    
    * enlarge atol for test_resnet_pure_fp16
    
    * refine code and enlarge atol
    
    * make custom_white_list and custom_black_list take effect for AMP and pure fp16
    
    * check tracer is not None
    
    * use default atol
    
    * change filter_size
    
    * change atol and add some NOTE
    52edad6a
test_resnet_pure_fp16.py 4.3 KB