Update gradient_accumulation_scheduler.py

ananthsub · Sep 19, 2020 · c470b97 · c470b97
1 parent d6df3c7
commit c470b97
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pytorch_lightning/callbacks/gradient_accumulation_scheduler.py b/pytorch_lightning/callbacks/gradient_accumulation_scheduler.py
@@ -56,7 +56,9 @@ def __init__(self, scheduling: dict):
 
         minimal_epoch = min(scheduling.keys())
         if minimal_epoch < 0:
-            raise IndexError(f"Epochs indexing from 1, epoch {minimal_epoch} cannot be interpreted correct")
+            raise IndexError(
+                f"Epochs indexing from 1, epoch {minimal_epoch} cannot be interpreted correct"
+            )
         if minimal_epoch != 0:  # if user didnt define first epoch accumulation factor
             scheduling.update({0: 1})