def METHOD_NAME(self, epoch, logs=None):
    """
    Run on end of each epoch
    """
    if logs is None:
        logs = dict()
    logger.debug(logs)
    # TensorFlow 2.0 API reference claims the key is `val_acc`, but in fact it's `val_accuracy`
    if 'val_acc' in logs:
        nni.report_intermediate_result(logs['val_acc'])
    else:
        nni.report_intermediate_result(logs['val_accuracy'])