It seems the code is failing with:
TypeError: can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.

while trying to call .numpy() on a tensor, which is still on the GPU, so you might need to move it to the CPU first.
You should also consider fixing this warning:
UserWarning: Using a target size (torch.Size([4208, 1])) that is different to the input size (torch.Size([4208, 75])). This will likely lead to incorrect results due to broadcasting. Please ensure they have the same size.