Model predicting robot offsets not learning

Question

I am currently trying to predict a serial robot\'s offsets by using a neural network on Pytorch. The method is the following :

• Sampling random a vector of joint