Hi @apaszke, thanks, but just a few questions - I currently have a class where I define my net: So I have:

class DNN(nn.Module):
def __init__(self):
super(DNN, self).__init__()



//stuff



def forward(self, input):


Now in my main script, I call:

myDNN = DNN()
myDNN.cuda()

I took a look at the documentation for the DataParallel, and what I currently do is this:

myDNN = torch.nn.DataParallel(DNN(), device_ids=[0, 1, 2])

Now this seems to run, BUT, it complains that it doesnt know what “forward” is. (The forward member function that it).
So I am confused as to what exactly I should be putting through this “DataParallel” exactly… every member function of my DNN?..
Thanks