train_curve.py [164:175]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
def train(
    model, train_loader, optimizer, criterion, epoch, device, **regime_params
):

    model.zero_grad()
    model.train()
    avg_loss = 0.0

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



train_indep.py [179:189]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
def train(
    model, train_loader, optimizer, criterion, epoch, device, **regime_params
):
    model.zero_grad()
    model.train()
    avg_loss = 0.0

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