I get this error when i run training with mulitiGPUs: nvalidArgumentError: 2 root error(s) found.
** (0) Invalid argument: Cannot batch tensors with different shapes in component 0. First element had shape [60,400,400,3] and element 7 had shape [32,400,400,3].**
Hi @Awatef_Edhib. Welcome to the forum.
Did you manage to get your code working on a CPU?
I mean the error message doesn’t suggest this relates to GPU/multi GPUs. It looks like a “regular” op issue due to different batch size. I may be wrong though.
Hi @Awatef_Edhib, The error occurs due to combing of 2 different batch size tensors. You can try by unbatching them. Also as @tagoma suggested you can run your code on single cpu to know if it is a problem occurs using multi GPU or problem in the code implementation. Thank You.