[train][data] Fix iter_torch_batches usage of ray.train.torch.get_device when running outside Ray Train
#57816
Loading
ray.train.torch.get_device when running outside Ray Train
#57816