Function unwrap_model

utils.py:116–119

Remove the DistributedDataParallel wrapper if present.

(model)

    amp = None

def unwrap_model(model):
    """Remove the DistributedDataParallel wrapper if present."""
    wrapped = isinstance(model, torch.nn.parallel.distributed.DistributedDataParallel)
    return model.module if wrapped else model


def load_checkpoint(config, model, optimizer, lr_scheduler, logger, model_ema=None):