Can I use torch.nn.DataParallel wrapped around my model after making a collate function to tackle variable sized inputs??
If yes how? Because when using a custom collate function like this 122 I am unable to run my model on multiple GPUs