[Feature Request] Cleaner interface for intermediate layers

[j29scott]

Would it be possible to get a cleaner interface for accessing bounds in intermediate layers? This is an extremely useful feature.

The ordered dictionary in _modules of a BoundedModule obtained from compute_bounds overwrites original names.

There is an .ori_name attribute, but its often missing. My solution is very hacky 😜

[huanzhang12]

Thanks for the suggestion. This is a problem on our list and we are currently working on it. We hope to get a better API for retrieving the intermediate layer bounds in the next release.

[j29scott]

Hi, any update on this?

Would be nice before the competition :)

[j29scott]

bumping! :)

[huanzhang12]

Sorry, none of the people in the team is available for making this change in the short term. If you want to implement this, I am happy to review your pull request.