Print weights summary
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?
just do
model = YourLightningModule(...)
model.summarize(mode=...)
mode='top', 'full'