This commit is contained in:
Oliver 2025-04-28 22:48:45 +01:00
parent fa52788e31
commit 2dff704f0b

View file

@ -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