def freeze_pretrained_model_params(pretrained_model):
    """ Freezes all pre-trained model params

        :param pretrained_model: a pre-trained model

        :return: same provided model with frozen parameters
    """
    for param in pretrained_model.parameters():
        param.requires_grad = False
    return pretrained_model