RefineRank / checkpoints /README.md
linzher's picture
Upload folder using huggingface_hub
4a24473 verified
|
Raw
History Blame Contribute Delete
1.12 kB
# Checkpoints
RefineRank 推理/训练所需的全部权重,平铺结构如下:
```
checkpoints/
├── vlm/ # 冻结的 MedVLM(Qwen2.5-VL 架构,HF 格式,全部文件直接放这里)
│ ├── config.json / model-0000X-of-00004.safetensors / tokenizer.json / ...
├── grounding_dino/ # 冻结的 GroundingDINO 权重
│ └── groundingdino_swinb_cogcoor.pth
└── refinenet/ # 唯一可训练模块 RefineNet(约 1.25M 参数)
├── proposal_adapter_full.pt # 随论文发布的最终提交权重
└── deployment_manifest.json
```
- 从 Hugging Face 一键恢复本目录:`hf download linzher/RefineRank --local-dir .`
- `vlm/``grounding_dino/` 在训练与推理中都保持冻结,仅用于特征与候选框提取。
- `python interface.py predict` 不带 `--checkpoint` 时优先使用本目录的平铺权重;
训练新产生的 `run_<时间戳>/` 目录(`python interface.py train` 的输出)也会被自动发现。
- 大权重文件不要提交到 git。