How to use DataParallel with custom classes?

12

Hey @michaelklachko ,
Thanks for the prompt reply, but I am unable to train my code on multiple GPUs,
It would be great if you could have a look at this thread: DataParallel not working as expected