Open TylerYep opened 4 years ago
HI @TylerYep, I've been one of the maintainers of this project, and now currently have some time to be able to devote to this. If you're interested, I can work with you to merge the projects.
That would be great! What would be the best way to discuss this?
Slack is the main platform comes to mind, though I'm open to any alternative suggestions!
Sounds good - could you send me a link? My email is on my profile page.
Sent, let me know if you receive it!
any news?
I never received write access to this project and, based on the implementation, am not too enthusiastic about handling backwards-incompatible changes in the new rewrite. Simply releasing a new version without a proper plan could break existing code in over 1400 codebases (based on GitHub). Additionally, torch-summary has deviated enough that I no longer regard it as a fork of this project - there is almost no overlap in code anymore.
I would recommend using my version if you want bug fixes and slightly more maintainability for the foreseeable future. If we gain a solid team of contributors we can revisit the merge.
Since this project hasn't been worked on in a few months, I decided to do a rewrite of the fixing a lot of the changes.
The changes are available on my fork and is installable using:
pip install torchinfo
https://github.com/TylerYep/torchinfo No changes to any of your code! It all works the same way but also handles specifying columns and handles LSTMs and RNNs as input, and incorporates @sangyx branching structure as output.
I plan on merging these two projects in the coming months (waiting on @sksq96 ), however if you need solutions to that problem soon you can just use my fork in the meantime.