If I use the following code for sorting batches will I face issues while running on single/multiple GPUs? If so Could you please explain why.

Screen Shot 2017-11-09 at 10.39.46 PM.png1374×420 48.1 KB