- 18 1月, 2021 1 次提交
-
-
由 Zhen Wang 提交于
-
- 04 12月, 2020 1 次提交
-
-
由 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
-
- 22 10月, 2020 1 次提交
-
-
由 Zhang Ting 提交于
-
- 19 5月, 2020 1 次提交
-
-
由 Zhen Wang 提交于
-
- 02 12月, 2019 1 次提交
-
-
由 ruri 提交于
-
- 26 11月, 2019 1 次提交
-
-
由 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
-