def get_padding_prp(hparams):
    """Get padding for pad_rotate_project measurements"""
    if hparams.dataset == 'mnist':
        paddings = [[0, 0], [6, 6], [6, 6], [0, 0]]
    elif hparams.dataset == 'celebA':
        paddings = [[0, 0], [14, 14], [14, 14], [0, 0]]
    else:
        raise NotImplementedError
    return paddings