I use BCELoss but the output of my model is tensor.FloatTensor, how to change it?
the output of your model needs to be Variable, not FloatTensor
1 Like
Thanks for your help!