## Description <!-- concise description of the feature/enhancement --> Add CPU ImageNet or MNIST example ## Motivation/Background <!-- why is this feature/enhancement important? provide background context --> Now we only have GPU ImageNet example, which requires multiple GPUs to test the feature. It is better to support CPU. I hacked the GPU ImageNet example to support CPU locally. I'd appreciate if we can support it in this repo.
Description
Add CPU ImageNet or MNIST example
Motivation/Background
Now we only have GPU ImageNet example, which requires multiple GPUs to test the feature. It is better to support CPU.
I hacked the GPU ImageNet example to support CPU locally. I'd appreciate if we can support it in this repo.