Pataki_Marton:

Can I use nn.DataParallel without using a Model class ?


Yes you can  See Uneven GPU utilization during training backpropagation 20 for an example wrapping the loss function with DataParallel