Pl_module vs trainer.model in Callbacks

callbacks
1

In callbacks, we can use either pl_module or trainer. model to access the actual lightning model that does the job. What is the difference?

2

pl_module is your LightningModule instance, while trainer.model can be an instance that wraps your LightningModule in certain strategies. For example, if you use DDP,

  • pl_module is an instance of LightningModule, and
  • trainer.model is DistributedDataParallel.

https://lightning.ai/docs/pytorch/stable/api/lightning.pytorch.trainer.trainer.Trainer.html#lightning.pytorch.trainer.trainer.Trainer.model

1 Like