Dataparallel size does not match with target

[gcramer23 (Garrett Cramer)]

Hi can you try adding some debugging logs to check your tensor shapes.