RichModelSummary¶
- class lightning.pytorch.callbacks.RichModelSummary(max_depth=1, **summarize_kwargs)[source]¶
Bases:
ModelSummary
Generates a summary of all layers in a
LightningModule
with rich text formatting.Install it with pip:
pip install rich
from lightning.pytorch import Trainer from lightning.pytorch.callbacks import RichModelSummary trainer = Trainer(callbacks=RichModelSummary())
You could also enable
RichModelSummary
using theRichProgressBar
from lightning.pytorch import Trainer from lightning.pytorch.callbacks import RichProgressBar trainer = Trainer(callbacks=RichProgressBar())
- Parameters:
- Raises:
ModuleNotFoundError – If required rich package is not installed on the device.