finished callbacks

pytorch_lightning/callbacks/pt_callbacks.py

@@ -1,15 +1,13 @@
 """
 Callbacks
-=========
-
+====================================
 Callbacks supported by Lightning
 """
 
-import logging
 import os
 import shutil
+import logging
 import warnings
-
 import numpy as np
 
 from pytorch_lightning.overrides.data_parallel import LightningDistributedDataParallel
@@ -165,7 +163,9 @@ class EarlyStopping(Callback):
 
 
 class ModelCheckpoint(Callback):
-    r"""Save the model after every epoch.
+    r"""
+
+    Save the model after every epoch.
 
     Args:
         filepath (str): path to save the model file.
@@ -403,4 +403,4 @@ class GradientAccumulationScheduler(Callback):
 #         should_stop = c.on_epoch_end(i, logs={'val_loss': loss})
 #         logging.info(loss)
 #         if should_stop:
-#             break
+#             break