Hi
That works thanks for all the help.
On a quick side questions, if I want to test my model with another set of x points, is this the right code: model(torch.tensor(x_points).to(device)).cpu().detach().clone().numpy().
Also how would you recommend improving the model.
Thanks