r.cuda() ?
Closed this issue · 1 comments
sk-g commented
YuxianMeng commented
@sk-g Should use r.cpu(). since the next line is not supported by gpu Tensor
Closed this issue · 1 comments
@sk-g Should use r.cpu(). since the next line is not supported by gpu Tensor