• W
    [Feature] Fp16 training for resnet50 (#14850) · 3d750f9c
    Wu Yi 提交于
    * wip
    
    * wip
    
    * wip
    
    * wip for test
    
    * add fp16 tests test=develop
    
    * fix cpu build test=develop
    
    * fix test=develop
    
    * fix py3 tests test=develop
    
    * fix lr_scheduler dtype test=develop
    
    * fix test=dvelop
    
    * test fix ci compile test=develop
    
    * fix build and merge test=develop
    
    * fallback momentumop change to general test=develop
    3d750f9c
top_k_op.cu 11.2 KB