YuxianMeng/Matrix-Capsules-pytorch

r.cuda() ?

Closed this issue · 1 comments

YuxianMeng commented

@sk-g Should use r.cpu(). since the next line is not supported by gpu Tensor

  • 👍1