• 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
train.py 8.9 KB