def get_model_device(model):
    """Get device on which model in situated.

    Parameters
    ----------
    model: nn.Module

    Returns
    -------
    device: torch.device
    """
    return model.parameters().__next__().device