lianjunw / dali_pytorch_demo Goto Github PK
View Code? Open in Web Editor NEWThis project forked from yaysummeriscoming/dali_pytorch_demo
Example code showing how to use Nvidia DALI in pytorch, with fallback to torchvision. Contains a few differences to the official Nvidia example, namely a completely CPU pipeline & improved memory usage
License: Apache License 2.0