Downloads · 30 days
0
deepswe-prm-heads-8k
tarsur385/deepswe-prm-heads-8k
deepswe-prm-heads-8k is a machine learning model from tarsur385. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for pytorch. The card lists the license as mit.
Three CLM projection-head checkpoints (statehead / actionhead / logitscale / cfg), fine-tuned with train/finetuneprmhead.py from jackyk02/qwen38bhead qwen38bmidtrainedhead.pt on tarsur385/deepswe-prm-train-embeddings-8k.
Access
Public
Updated Sep 21, 2026
Repo size
227 MB
Likes
1
Public
Repo makeup
Click a slice to open those files.
.pt227 MB · 100%
At a glance
- Library
- pytorch
- License
- mit
- Access
- Public
- Created
- Sep 21, 2026
- Updated
- Sep 21, 2026
- SHA
- c49bf9c1
Model card
From the Hugging Face model README
DeepSWE PRM heads (Qwen3-8B, 8k)
Three CLM projection-head checkpoints (state_head / action_head / logit_scale / cfg),
fine-tuned with train/finetune_prm_head.py from
jackyk02/qwen3_8b_head qwen3_8b_midtrained_head.pt
on tarsur385/deepswe-prm-train-embeddings-8k.
Head fold{k} excludes the tasks in folds/fold{k}.json; fold_spec.json maps each head to those tasks.
Usage
From the root of contrastive_learning:
hf download tarsur385/deepswe-prm-heads-8k --local-dir heads
python evaluation/bon_deepswe_eval.py --hf-dataset tarsur385/deepswe-prm-embeddings-8k \
--fold-spec heads/fold_spec.json
Training
python train/finetune.py --task prm --hf-dataset tarsur385/deepswe-prm-train-embeddings-8k \
--init-ckpt qwen3_8b_midtrained_head.pt --out-dir runs/deepswe \
--holdout-folds heads/folds/fold0.json heads/folds/fold1.json heads/folds/fold2.json
- Library
- pytorch
- License
- mit
- Created
- Sep 21, 2026
- Updated
- Sep 21, 2026