Closed blackyang closed 7 years ago
If have torch.set_default_tensor_type('torch.cuda.FloatTensor') somewhere else, there will be a data type mismatch error (which too me a lot of time to debug lol). Now force costs to be a cpu tensor.
torch.set_default_tensor_type('torch.cuda.FloatTensor')
thanks!
SeanNaren
commented
7 years ago
If have
torch.set_default_tensor_type('torch.cuda.FloatTensor')somewhere else, there will be a data type mismatch error (which too me a lot of time to debug lol). Now force costs to be a cpu tensor.