Recursive Summary for Models
🚀 Feature
At the moment whenever you start training you get a print out of the model's modules and number of parameters. It would be great if you were able to recursively traverse modules and print out the modules and parameters in these submodules. This could be enabled through a possible flag verbose, which would further print out the recursive definitions.
Currently:
  | Name       | Type            | Params
-----------------------------------------------
0 | features   | Motion_Features | 34 K
1 | classifier | Prediction_Head | 2 M

Future:
   | Name  | Type      | Params
-------------------------------------
0 | features   | Motion_Features | 34 K
- 0  | conv1 | Conv2d    | 8 K
- 1  | conv2 | Conv2d    | 13 K
- 2  | mp1   | MaxPool2d | 0
- 3  | conv3 | Conv2d    | 4 K
- 4  | conv4 | Conv2d    | 3 K
- 5  | mp2   | MaxPool2d | 0
- 6  | conv5 | Conv2d    | 3 K
- 7  | conv6 | Conv2d    | 580
1 | classifier | Prediction_Head | 2 M
- 8  | lin1  | Linear    | 2 M
- 9  | head1 | Linear    | 20 K
- 10 | head2 | Linear    | 20 K
- 11 | out1  | Linear    | 1 K
- 12 | out2  | Linear    | 101


Motivation
This would enable user to easily see all of the layers they are using and the total number of parameters being used for all network components.
Pitch
I would like someone to rewrite summarize to have a verbose flag to further print out each of the submodules and number of parameters.