diff --git a/pytorch_lightning/callbacks/pt_callbacks.py b/pytorch_lightning/callbacks/pt_callbacks.py index 001981e4..1549f54d 100644 --- a/pytorch_lightning/callbacks/pt_callbacks.py +++ b/pytorch_lightning/callbacks/pt_callbacks.py @@ -382,6 +382,7 @@ class GradientAccumulationScheduler(Callback): if minimal_epoch < 1: msg = f"Epochs indexing from 1, epoch {minimal_epoch} cannot be interpreted correct" raise IndexError(msg) + elif minimal_epoch != 1: # if user didnt define first epoch accumulation factor scheduling.update({1: 1})