Hi, I am not sure, but probably here you don’t need to put batch on device twice.



 TeaWaterSleep:

batch = batch.to(device)
        encod_decod = autoencod(batch.to(device))



Also at this line your model (autoencod) is on device, but data is not



 TeaWaterSleep:

t_l = loss(dataset.data, autoencod(dataset.data))