From 55a305e43c7a77af0434848877e14a7ba0c79ddf Mon Sep 17 00:00:00 2001
From: William Falcon <waf2107@columbia.edu>
Date: Wed, 15 Jan 2020 20:36:57 -0500
Subject: [PATCH] finished callbacks

---
 pytorch_lightning/callbacks/pt_callbacks.py | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/pytorch_lightning/callbacks/pt_callbacks.py b/pytorch_lightning/callbacks/pt_callbacks.py
index 42a8336f..001981e4 100644
--- a/pytorch_lightning/callbacks/pt_callbacks.py
+++ b/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
\ No newline at end of file