diff --git a/pytorch_lightning/callbacks/pt_callbacks.py b/pytorch_lightning/callbacks/pt_callbacks.py index 70da98d2..71e61f9b 100644 --- a/pytorch_lightning/callbacks/pt_callbacks.py +++ b/pytorch_lightning/callbacks/pt_callbacks.py @@ -361,6 +361,7 @@ class GradientAccumulationScheduler(Callback): Args: scheduling (dict): scheduling in format {epoch: accumulation_factor} + Example:: from pytorch_lightning import Trainer