I am working on a model with data larger than RAM, so I have to use \'generator\' way instead of \'all-saved-to-RAM\' way. Recently, we are trying to migrate to multi-GPU ma