Hi, tensor on your GPU is not torch.tensor.
So you need to use .cpu()
before you call .numpy()
.
Please check related topic for detail:
Shortly, it would be like:
... data.cpu().numpy()
Hi, tensor on your GPU is not torch.tensor.
So you need to use .cpu()
before you call .numpy()
.
Please check related topic for detail:
Shortly, it would be like:
... data.cpu().numpy()