code: def on_train_begin(self, logs=None):
      self.flops = get_flops()
  
file_path: auptimizer/Examples/early_stopping/mnist_keras/mnist.py
imports: import aup
  import tensorflow
  import numpy
  from tensorflow import keras
  from tensorflow.keras import layers
  from math import log
  import sys
label_desc: Use image node_id 2 for calling the CustomCallback obj's underlying member method code with example usage: obj.on_train_begin(logs) without return types