1. 24 1月, 2019 1 次提交
  2. 10 1月, 2019 1 次提交
    • W
      [Feature] support mix precision training for resnet (#14899) · fd854183
      Wu Yi 提交于
      * clip softmax for fp16
      
      * updates
      
      * fuse xent support fp16 test=develop
      
      * wip
      
      * wip
      
      * add simple row reduce
      
      * wip fp16 accurate softmax
      
      * add accurate softmax kernel for fp16 test=develop
      
      * update test=develop
      
      * fix cpu build test=develop
      
      * update api.spec test=develop
      
      * follow comments test=develop
      
      * fix build test=develop
      
      * fix trt build test=develop
      
      * fix inference build test=develop
      
      * fix merge test=develop
      
      * update test=develop
      
      * try fix build test=develop
      
      * fix build test=develop
      
      * rename real_exp test=develop
      
      * fortest
      
      * remove hacky kernels test=develop
      
      * clean up test=develop
      fd854183
  3. 26 12月, 2018 1 次提交
    • W
      Fp16 training (#14992) · 856f0da0
      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
      
      * make fp16 lr schedule simple test=develop
      
      * fix ut test=develop
      
      * fix tests test=develop
      
      * remove fp16 learning rate cast test=develop
      856f0da0
  4. 24 12月, 2018 1 次提交
  5. 21 12月, 2018 2 次提交
  6. 20 12月, 2018 3 次提交
  7. 19 12月, 2018 5 次提交
  8. 18 12月, 2018 5 次提交
  9. 17 12月, 2018 1 次提交
  10. 12 12月, 2018 2 次提交
  11. 11 12月, 2018 2 次提交
  12. 05 12月, 2018 1 次提交
    • C
      Fix clip.py (#14718) · 04539d4c
      chengduo 提交于
      * expose square
      test=develop
      
      * fix activation
      test=develop
      
      * Add square API
      test=develop
      
      * add necessary op
      
      * code refine
      
      * fix API.spec
      test=develop
      
      * fix unit test
      test=develop
      
      * add unit test sparse_grad_clip
      test=develop
      
      * fix API.spec
      test=develop
      
      * remove mac test for test_gradient_clip
      test=develop
      
      * remove selectedrows_mul_tensor
      test=develop
      04539d4c
  13. 27 11月, 2018 1 次提交
  14. 16 11月, 2018 4 次提交