RAM overloading compared to VRAM

[dsmertin]

I'm trying to launch MipNerf traing but like in #12 I have huge CPU memory consumption while GPU idles. I guess it's because data loader and sampler working on CPU.
Is there a way to move this processing to GPU?