I am a beginner with PyTorch. I want to identify the matrix A in the ODE: dX/dt=A*X. So, I wrote a PyTorch code to do that. I only use a linear cell in the neural network.
Unfortunately for me, it doesn’t work at all. When I run the code, the loss is decreasing, but when I check the weights of the matrix, they are totally different from the coefficients of the matrix A.
Can someone help me?
The link of the notebook : https://colab.research.google.com/drive/1hngoL3R9G7jx3Yv-mTYMGRPFjbHiaaCC?usp=sharing