and most tutorials  I read use the slow Dataloader  solution whereas using something like:
 for i in range((n - 1) // bs + 1):
`   x = x_ts[i * bs: i * bs + bs]`     # x_ts  in GPU

makes it much faster (with one worker of course).