yuxihu opened a new pull request #14771: [v1.4.x] Add pin_device_id option to Gluon DataLoader (#14136) URL: https://github.com/apache/incubator-mxnet/pull/14771 Cherry-pick the change that fixes out of memory error when using Gluon DataLoader with CPUPinned memory during distributed training on multiple GPUs with MXNet + Horovod to v1.4.x branch.
---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: [email protected] With regards, Apache Git Services
