ModelSummary¶
- class pytorch_lightning.callbacks.ModelSummary(max_depth=1)[source]¶
Bases:
pytorch_lightning.callbacks.base.Callback
Generates a summary of all layers in a
LightningModule
.- Parameters
max_depth¶ (
int
) – The maximum depth of layer nesting that the summary will include. A value of 0 turns the layer summary off.
Example:
>>> from pytorch_lightning import Trainer >>> from pytorch_lightning.callbacks import ModelSummary >>> trainer = Trainer(callbacks=[ModelSummary(max_depth=1)])