Solved by JuanFMontesinos in post #2 
              
              
                Resizing your images is kind of “mandatory”. You would need to pad them otherwise. It’s up to you and there are several ways in the literature. 
For sure, Dataparallel requires a tensors of same size. 
If you are using pytorch dataloader you can wheter to preprocess your dataset or to apply resizing…