goknur arican

Results 2 comments of goknur arican

model.train(device='cuda') is not working: RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!

what are you working on

MNIST Example

I trained a ClassicMLP model on MNIST with an input size of 784, hidden layer sizes of [128, 64], and an output size of 10. This model was trained using...