Show summary up to error

[R-N]

Is your feature request related to a problem? Please describe.
Torchinfo summary is very useful to debug shapes and module chain. However, it doesn't work when it encounters error. Often, errors are caused by wrong shape. This is where torchinfo summary should be very useful.

Describe the solution you'd like
When summary encounters error, print summary up to the error layer.

Describe alternatives you've considered
print manually