Pytorch Lightening Refactoring

[vincentchoqueuse]

Hi KalmanNet user,

Thank you for your interesting repo and research project on Kalman Net. To improve the reusability of your repo, maybe it could be a good idea to use the pytorch lightening module (https://github.com/PyTorchLightning/pytorch-lightning). I think it could help other users to dive into your code.

[XiaoyongNI]

Hi Prof. Choqueuse,

Thanks for your advice!

[SongJgit]

I also have this idea, have you finished refactoring？

[SongJgit]

I have finished refactoring this project using pl and will release the code after the paper is published