Jesús Miguel Rojas Gómez

icon location Granada icon location Currently studying the Dual Bachelor in Mathematics and Computer Science at University of Granada.

Results 1 comments of Jesús Miguel Rojas Gómez

Expected a 'cuda' device type for generator but found 'cpu'

I got the same error and I fixed it changing ....../torch/utils/data/sampler.py line 121: generator = torch.Generator() to generator = torch.Generator(device='cuda') (using for example vi editor). Then if you want to...