I’m getting the object is not having .cuda() method when using the second one. From getitem I’m returning a tuple of (image, label). I have successfully used transformations, changed the dtype to float32. But it’s outputting the error .cuda() method can be used on that.
But works totally fine by 1st method.