Modify param printing to show top level modules only
Add a model summary option that just shows the number of parameters for the top level modules.

Set pd.options.display.float_format = ’{:,}’.for- mat so it uses comma separators.
Also summarize the number of parameters like 1,000,234 to 1M.
use 1k, 1M, 1B, 1T, etc...

Suggested by @adefazio
Specifically, change:
Trainer(print_weights_summary=True)
To:
# default
Trainer(weights_summary='full')

# second option
Trainer(weights_summary='top')