grp_reranker_best.pt — reranker weights + 37 feature names + feat_mean/feat_std (self-contained normalization).modeling_reranker.py — GroupReranker (the head) + GapPredictorV13 (Stage-1 base) source.1import torch
2from modeling_reranker import GroupReranker
3obj = torch.load('grp_reranker_best.pt', map_location='cpu', weights_only=False)
4feats = obj['features'] # 37 feature names, in order
5m = GroupReranker(len(feats), 1024, 256, 512, 4, 320, 8, 0.12)
6m.load_state_dict(obj['model'], strict=True); m.eval()GapPredictorV13 + best_v15.pt → top-10 candidate gaps + 1024-d hidden vectors.GroupReranker, argmax over candidates.