We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 24cc8d8 commit c71488fCopy full SHA for c71488f
docs/source/common/debugging.rst
@@ -103,6 +103,7 @@ By default it only prints the top-level modules. If you want to show all submodu
103
104
trainer = Trainer(callbacks=[ModelSummary(max_depth=-1)])
105
106
+
107
You can also display the intermediate input- and output sizes of all your layers by setting the
108
``example_input_array`` attribute in your LightningModule. It will print a table like this
109
0 commit comments