Hi, Thanks for the response
What do you mean by “push it to the GPU beforehand…”? do you mean to simply call  .cuda() on the loaded training data and labels and to simply go over this tensor in batches?