Custom Batch class won't send to the correct device

DDP/GPU
2

hey @artuntun

you might want to look at transfer_batch_to_device hook.

Also, we have moved the discussions to GitHub Discussions. You might want to check that out instead to get quick response. The forums will be marked read-only after some time.

Thank you