Class CodeCarbonCallBack

examples/mnist_callback.py:12–17

class CodeCarbonCallBack(Callback):
    def __init__(self, codecarbon_tracker):
        self.codecarbon_tracker = codecarbon_tracker

    def on_epoch_end(self, epoch, logs=None):
        self.codecarbon_tracker.flush()


mnist = tf.keras.datasets.mnist