Skip to content
  • Indrayana Rustandi's avatar
    Add Dali MNIST example (#3721) · 6e5f232f
    Indrayana Rustandi authored
    
    
    * add MNIST DALI example, update README.md
    
    * Fix PEP8 warnings
    
    * reformatted using black
    
    * add mnist_dali to test_examples.py
    
    * Add documentation as docstrings
    
    * add nvidia-pyindex and nvidia-dali-cuda100
    
    * replace nvidia-pyindex with --extra-index-url
    
    * mark mnist_dali test as Linux and GPU only
    
    * adjust CUDA docker and examples.txt, fix import error in test_examples.py
    
    * adjust the GPU check
    
    * Exit when DALI is not available
    
    * remove requirements-examples.txt and DALI pip install
    
    * Refactored example, moved to new logging api, added runtime check for test and dali script
    
    * Patch to reflect the mnist example module
    
    * add req.
    
    * Apply suggestions from code review
    
    * Removed requirement as it breaks CPU install, added note in README to install DALI
    
    * add DALI to Drone
    
    * test examples
    
    * Apply suggestions from code review
    
    * imports
    
    * ABC
    
    * cuda
    
    * cuda
    
    * pip DALI
    
    * Move build into init function
    
    Co-authored-by: default avatarSeanNaren <sean@grid.ai>
    Co-authored-by: default avatarJirka Borovec <jirka@pytorchlightning.ai>
    Co-authored-by: default avatarJirka Borovec <Borda@users.noreply.github.com>
    Co-authored-by: default avatarSean Naren <sean.narenthiran@gmail.com>
    6e5f232f