in src/run_paraphrase.py [0:0]
def prune_features(feat_dicts): return [{k: v for k, v in d.items() if int(k.split('-')[1]) in FEATURE_LAYERS} for d in feat_dicts]