def get_model_ema_state(model):
    """Return the ema state stored in `model`"""
    model = _remove_ddp(model)
    assert hasattr(model, "ema_state")
    ema = model.ema_state
    return ema
Return the ema state stored in `model`
EVA-02/det/detectron2/modeling/ema.py