@shashimalcse
The way you have written the code your model’s parameter won’t update, since you are detaching the xyzk which means the gradient won’t flow past it. Rather than converting xyzk to a numpy array, keep it as a torch.Tensor & change the other variables to a torch.Tensor too.