RuntimeError: Can't call numpy() on Tensor that requires grad. Use tensor.detach().numpy() instead.
TypeError: Cannot interpret 'torch.float32' as a data type
RuntimeError: Can't call numpy() on Tensor that requires grad. Use tensor.detach().numpy() instead.
TypeError: Cannot interpret 'torch.float32' as a data type