From 150a94543c2ac346bd220bfc2b5c2dcc8e5b110d Mon Sep 17 00:00:00 2001 From: William Falcon Date: Thu, 16 Jan 2020 16:50:05 -0500 Subject: [PATCH] finished lightning module --- pytorch_lightning/callbacks/pt_callbacks.py | 1 + 1 file changed, 1 insertion(+) 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