Bug fixes for nested modules.

jacobkimmel / pytorch_modelsize

Estimates the size of a PyTorch model in memory

MIT License

354 stars 49 forks source link

Closed shijianjian closed 4 years ago

andimarafioti commented 4 years ago

This didn't work for me, the function 'get_output_sizes_from_module' is never called outside of it's own definition.

jacobkimmel commented 4 years ago

Agreed @andimarafioti, this PR has not been merged since the function is never used. I'll close this because it's stale now.