How to use torch nn.DataParallel in R

Machine Learning and Modeling
1

I've come across the DataParallel class in torch.nn here. However it doesn't seem to be part of the R torch package, and I can't find it anywhere in the documentation. Is this something that is going to be implemented in the future?

2

Hi @chaz ,

Yes, this not yet implemented in the R package and we plan to implement this feature in future versions of torch. I just opened an issue in the GitHub repository so we can track development: Implement data parallelism for nn_modules · Issue #573 · mlverse/torch · GitHub