Hello @ptrblck , thanks for the response!
Could you please help me with clues on how to debug this?
What I don’t understand is, why is this error being triggered only when I wrap the model under nn.Dataparallel ? and not when I run inference on a single GPU.
If it is of any help, my only call to put the model for multi-GPU usage is a simple conditional:
if torch.cuda.device_count()>1:
    # Multi-GPU Training
    model = nn.DataParallel(model)

Thank you for any help that can be provided.