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()