model_logging.py [52:61]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    model,
    loss,
    acc,
    model_type,
    param=None,
    metric_dict=None,
    extra_metrics=None,
    **regime_params,
):
    metric_dict = _handle_bn_stats(metric_dict, extra_metrics, regime_params)
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



model_logging.py [82:91]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    model,
    loss,
    acc,
    model_type,
    param=None,
    metric_dict=None,
    extra_metrics=None,
    **regime_params,
):
    metric_dict = _handle_bn_stats(metric_dict, extra_metrics, regime_params)
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -