search

KalmanNet / KalmanNet_TSP

code for KalmanNet
287 stars 76 forks source link

Pytorch Lightening Refactoring #4

Open vincentchoqueuse opened 2 years ago

vincentchoqueuse commented 2 years ago

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 commented 2 years ago

Hi Prof. Choqueuse,

Thanks for your advice!

SongJgit commented 1 year ago

I also have this idea, have you finished refactoring？

SongJgit commented 1 year ago

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