def METHOD_NAME(self, model, net_output, sample, reduce=True):
    loss, _ = super().METHOD_NAME(model, net_output, sample, reduce)
    aux_loss = model.get_aux_loss()
    avg_span = model.get_current_avg_span()
    max_span = model.get_current_max_span()
    return loss, aux_loss, avg_span, max_span