chore: remove unused callback

mattmolinare · Jan 11, 2022 · 2809456 · 2809456
1 parent 997bdc5
commit 2809456
Showing 1 changed file with 0 additions and 17 deletions.
diff --git a/src/chiron/callbacks.py b/src/chiron/callbacks.py
@@ -291,20 +291,3 @@ def __init__(self, base_lr, max_lr, step_size, gamma):
 
     def get_scale(self, step, cycle):
         return self.gamma ** step
-
-
-class Counter(tf.keras.callbacks.Callback):
-    def __init__(self):
-        super().__init__()
-        self.epoch = None
-        self.batch = None
-
-    def on_train_begin(self, logs=None):
-        self.epoch = 0
-        self.batch = 0
-
-    def on_train_batch_end(self, batch, logs=None):
-        self.batch += 1
-
-    def on_train_epoch_end(self, epoch, logs=None):
-        self.epoch += 1