Open 0wwafa opened 3 months ago
According to stackoverflow:
You are using CPU only pytorch, but your code has statement like cr = nn.CrossEntropyLoss(weight=torch.tensor(classes_weights).cuda()) which is trying to move the tensor to GPU.
To fix it, remove all the .cuda() operations.
I did everything in the README. (I am using CPU only)
When running python app.py I get: AssertionError: Torch not compiled with CUDA enabled