ModelSummary¶

class lightning.pytorch.callbacks.ModelSummary(max_depth=1)[소스]¶

기반 클래스: lightning.pytorch.callbacks.callback.Callback

Generates a summary of all layers in a LightningModule.

매개변수: 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 lightning.pytorch import Trainer
>>> from lightning.pytorch.callbacks import ModelSummary
>>> trainer = Trainer(callbacks=[ModelSummary(max_depth=1)])

on_fit_start(trainer, pl_module)[소스]¶

Called when fit begins.

반환 형식: None