mirror of
https://github.com/open-thought/reasoning-gym.git
synced 2026-04-24 17:05:03 +00:00
cfg
This commit is contained in:
parent
fa52788e31
commit
2dff704f0b
1 changed files with 2 additions and 0 deletions
|
|
@ -94,6 +94,8 @@ actor_rollout_ref:
|
|||
ppo_epochs: 1
|
||||
shuffle: False
|
||||
ulysses_sequence_parallel_size: 1 # sp size
|
||||
checkpoint:
|
||||
contents: ['model', 'optimizer', 'extra']
|
||||
optim:
|
||||
lr: 1e-6
|
||||
lr_warmup_steps_ratio: 0. # the total steps will be injected during runtime
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue