open-thought/reasoning-gym
1
0
Fork 0
mirror of https://github.com/open-thought/reasoning-gym.git synced 2026-04-25 17:10:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

cfg

Browse source
This commit is contained in:
Oliver 2025-04-28 22:48:45 +01:00
parent fa52788e31
commit 2dff704f0b
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
training/configs/external_generalisation/math_curriculum_qwen_7b.yaml
View file

@ -94,6 +94,8 @@ actor_rollout_ref:
ppo_epochs: 1 ppo_epochs: 1
shuffle: False shuffle: False
ulysses_sequence_parallel_size: 1 # sp size ulysses_sequence_parallel_size: 1 # sp size
checkpoint:
contents: ['model', 'optimizer', 'extra']
optim: optim:
lr: 1e-6 lr: 1e-6
lr_warmup_steps_ratio: 0. # the total steps will be injected during runtime lr_warmup_steps_ratio: 0. # the total steps will be injected during runtime