How do I print output of each layer in sequential?

This can be achieved by forward hooks.

1 Like