TRAttack icon indicating copy to clipboard operation
TRAttack copied to clipboard
verified publisher icon amirgholami

add cpu(), fix can't convert bug

Open TYTaO opened this issue 5 years ago • 0 comments

can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.

TYTaO avatar Dec 11 '20 03:12 TYTaO