Hi,
Note that you should never use .data. You can use .detach() if you want a Tensor with the same content but that does not track gradients. Or with torch.no_grad(): if you want to locally disable the autograd.