Evangelinejy's picture
model_50m_0.69B card
906ae88 verified
|
Raw
History Blame Contribute Delete
1.37 kB
metadata
license: apache-2.0
tags:
  - chess
  - reinforcement-learning
  - grpo

model_50m_0.69B — RL (GRPO) checkpoints

RL post-training trajectory for the chess pre-to-post compute-allocation study. The pretraining base and SFT init for this model are model_50m_0.69B and model_50m_0.69B.

parameters 50m
pretraining tokens 687,325,398 (0.687B)
checkpoints here 12 (steps 100–1200)
checkpoints saved by the run 24

Steps

100, 200, 300, 400, 500, 600, 700, 800, 900, 1000, 1100, 1200

Loading

Each global_step_N/ folder is self-contained. The models use a custom tokenizer (tokenizer.py), and the remote-code resolver ignores subfolder=, so download the folder first and load the local path:

from huggingface_hub import snapshot_download
from transformers import AutoModelForCausalLM, AutoTokenizer

step = "global_step_1200"
p = snapshot_download("Pre2Post-Chess-RL/Chess-RL-Models", allow_patterns=f"model_50m_0.69B/{step}/*") + f"/model_50m_0.69B/{step}"
model = AutoModelForCausalLM.from_pretrained(p, trust_remote_code=True)
tok = AutoTokenizer.from_pretrained(p, trust_remote_code=True)