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

语言：Python · GitHub 星标：196 · Fork：20

在 GitHub 上查看原仓库