| #!/usr/bin/env python | |
| """GDPO 权重 probe: 打印键结构, 供 src/common.load_gdpo_teacher 排错。 | |
| 用法: python scripts/probe_gdpo.py weight/gdpo/ckp/diffusion_pytorch_model.safetensors | |
| """ | |
| import sys | |
| from pathlib import Path | |
| sys.path.insert(0, str(Path(__file__).resolve().parents[1] / "src")) | |
| from common import probe_gdpo | |
| if __name__ == "__main__": | |
| if len(sys.argv) < 2: | |
| print("用法: probe_gdpo.py <gdpo权重路径>"); sys.exit(1) | |
| probe_gdpo(sys.argv[1]) | |