Function unwrap

src/post_training/utils.py:94–96

Return the underlying module behind a DDP wrapper (or the model itself).

(model: nn.Module)

def unwrap(model: nn.Module) -> nn.Module:
    """Return the underlying module behind a DDP wrapper (or the model itself)."""
    return model.module if hasattr(model, "module") else model


def make_frozen_copy(model: nn.Module, device: str | None = None) -> nn.Module: