can't convert cuda tensor to numpy. use tensor.cpu() to copy the tensor
By Thankful Tarantula at Jun 12 2021