Closed doantientai closed 5 years ago
Based on the error message, it seems like you forgot to call .cuda for your tensor.
@doantientai
It seems to occured in Line 244 of utils.py
.
You should modify the code below.
batch = batch.sub(Variable(mean)) # subtract mean
to
batch = batch.sub(Variable(mean).cuda()) # subtract mean
Hi,
I am trying to implement the network on my own data-set, but received the error. I don't know where does it come from. Please help. Thank you a lot!