Can you paste the output for our_dataset.dtype and our_labels.dtype
And also did you send the data and labels to the GPU while training?
For example:
inputs, labels = data
inputs, labels = Variable(inputs).cuda(), Variable(labels).cuda()
our_dataset.dtype should be of type (np.float32)
our_labels.dtype should be of type ('int64')