Merge pull request #14 from achaiah/WIP

Wip

pywick/callbacks/CSVLogger.py

@@ -7,6 +7,7 @@
 
 from . import Callback
 
+__all__ = ['CSVLogger']
 
 class CSVLogger(Callback):
     """

pywick/callbacks/EarlyStopping.py

@@ -1,5 +1,7 @@
 from . import Callback
 
+__all__ = ['EarlyStopping']
+
 class EarlyStopping(Callback):
     """
     Early Stopping to terminate training early under certain conditions

pywick/callbacks/ExperimentLogger.py

@@ -6,6 +6,7 @@
 
 from . import Callback
 
+__all__ = ['ExperimentLogger']
 
 class ExperimentLogger(Callback):
     """

pywick/callbacks/History.py

@@ -1,5 +1,8 @@
 from . import Callback
 
+__all__ = ['History']
+
+
 class History(Callback):
     """
     Callback that records events into a `History` object.

pywick/callbacks/LRScheduler.py

@@ -2,6 +2,8 @@
 
 from . import Callback
 
+__all__ = ['LRScheduler']
+
 
 class LRScheduler(Callback):
     """

pywick/callbacks/ModelCheckpoint.py

@@ -8,6 +8,9 @@
 from . import Callback
 
 
+__all__ = ['ModelCheckpoint']
+
+
 class ModelCheckpoint(Callback):
     """
     Model Checkpoint to save model weights during training. 'Best' is determined by minimizing the value found under monitored_log_key in the logs

pywick/callbacks/OneCycleLRScheduler.py

@@ -11,6 +11,7 @@
 from . import Callback
 import torch
 
+__all__ = ['OnceCycleLRScheduler']
 
 class OneCycleLR(torch.optim.lr_scheduler.StepLR):
     r"""Sets the learning rate of each parameter group according to the

pywick/callbacks/ReduceLROnPlateau.py

@@ -1,5 +1,8 @@
 from . import Callback
 
+__all__ = ['ReduceLROnPlateau']
+
+
 class ReduceLROnPlateau(Callback):
     """
     Reduce the learning rate if the train or validation loss plateaus

pywick/callbacks/SimpleModelCheckpoint.py

@@ -5,6 +5,8 @@
 
 from . import Callback
 
+__all__ = ['SimpleModelCheckpoint']
+
 
 class SimpleModelCheckpoint(Callback):
     """

pywick/callbacks/TQDM.py

@@ -1,6 +1,9 @@
 from tqdm import tqdm
 from . import Callback
 
+__all__ = ['TQDM']
+
+
 class TQDM(Callback):
 
     def __init__(self):

pywick/callbacks/__init__.py

@@ -14,6 +14,7 @@
 from .LambdaCallback import *
 from .LRScheduler import *
 from .ModelCheckpoint import *
+from .OneCycleLRScheduler import *
 from .ReduceLROnPlateau import *
 from .SimpleModelCheckpoint import *
-from .TQDM import *
+from .TQDM import *