1. 18 1月, 2021 1 次提交
  2. 04 12月, 2020 1 次提交
    • Z
      Support pure fp16 training for ResNet50. (#4979) · ba1b19c0
      Zhen Wang 提交于
      * add pure fp16 training.
      
      * update the usage of cast_parameters_to_fp16.
      
      * use fp32 to initilize parameter values and then cast to fp16.
      
      * enable multi_precision when training with fp16.
      
      * add static loss scaling.
      
      * add the rescale_grad function in the pure fp16 training.
      
      * use cast to complete the pure fp16 training.
      
      * use sum loss for fp16 trainging and mean loss for fp32 training.
      
      * refine codes according reviewers' suggestions
      ba1b19c0
  3. 22 10月, 2020 1 次提交
  4. 19 5月, 2020 1 次提交
  5. 02 12月, 2019 1 次提交
  6. 26 11月, 2019 1 次提交
    • Y
      Add dali preprocessing pipeline for imagenet training (#3960) · 2e4465c6
      Yang Zhang 提交于
      * Use iterable dataloader for image classification
      
      * Support imagenet dataset with plain folder structure
      
      * Add dali section to README.md
      
      * Improve docs
      
      * Remove some flags from docs
      
      * Add dali reader for image classification
      
      * Fix elapsed time calculation
      2e4465c6