Open thatgeeman opened 2 years ago
If working with GPUs the model has to be moved to the current device, since the images and labels of the mini-batch is also being moved.
If working with GPUs the model has to be moved to the current device, since the images and labels of the mini-batch is also being moved.