search

atong01 / conditional-flow-matching

TorchCFM: a Conditional Flow Matching library
https://arxiv.org/abs/2302.00482
MIT License
875 stars 60 forks source link

Remove Pytorch Lightning? #121

Open GeoffStr opened 3 days ago

GeoffStr commented 3 days ago

Hi!

Great repo.

The only difficulty I have is that the pytorch lightning abstractions have made integrating into other projects pretty hard. I tend to think plain pytorch is actually cleaner and easier to understand these days -- as well as much more flexible :)! I would love to build on your work, but I'm a bit stuck with the pytorch lightning induced issues.

Do you have any plans on the roadmap to pull out the pytorch lightning related code? If so, I would be really interested in helping! Thanks!

atong01 commented 1 day ago

Hi!

This is not on my roadmap. Too busy! But I agree with you, it does make things more difficult to reuse. If there are specific things you think would be useful to pull out, happy to consider how to do this.