Print weights summary

[sho-87]

Weights summary gets printed when Trainer calls fit(), but the output is not persistent as I have everything wrapped up in ray tune, which overwrites the contents of the output cell in jupyter

Is there something we can call to manually print the weights summary of a particular model without having to fit the model every time?

[rohitgr7]

just do

model = YourLightningModule(...)
model.summarize(mode=...)

mode='top', 'full'