I am trying to train my model on multiple GPUs but I have some trouble with torch.distributions.Laplace that I call in the forward pass.
I have uploaded a minimal working example 6 that runs fine without torch.nn.DataParallel but fails when using it.
Is there any way to make this code run on multiple GPUs?