If I type:
print(our_dataset.dtype, our_labels.dtype)
I get:
float32 int64
I sent the data to cuda (and gave the Error in that case). However, if I don’t send the net (and data) to cuda, I get the other error that I posted in my original post.
@edgarriba
Well, the data is already in float.