train_curve.py [358:372]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        metric_dict=metric_dict,
        extra_metrics=extra_metrics,
        **regime_params,
    )


def train_model(config):
    # Get network, data, and training parameters
    (
        net,
        opt_params,
        train_params,
        regime_params,
        data,
    ) = utils.network_and_params(config)
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



train_indep.py [383:397]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        metric_dict=metric_dict,
        extra_metrics=extra_metrics,
        **regime_params,
    )


def train_model(config):
    # Get network, data, and training parameters
    (
        net,
        opt_params,
        train_params,
        regime_params,
        data,
    ) = utils.network_and_params(config)
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -