train_curve.py [291:307]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    val_loader,
    criterion,
    epoch,
    device,
    metric_dict=None,
    **regime_params,
):

    model.zero_grad()
    model.eval()
    test_loss = 0
    correct = 0

    regime_update_dict = {
        "sparse": sparse_module_updates,
        "quantized": quantized_module_updates,
        "lec": lec_update,
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



train_indep.py [314:330]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    val_loader,
    criterion,
    epoch,
    device,
    metric_dict=None,
    **regime_params,
):

    model.zero_grad()
    model.eval()
    test_loss = 0
    correct = 0

    regime_update_dict = {
        "sparse": sparse_module_updates,
        "quantized": quantized_module_updates,
        "lec": lec_update,
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -