Skip to content

Commit 6324f38

Browse files
authored
Add gspo-token to loss_config.method (#426)
add gspo-token to loss_config.method
1 parent 3ddc996 commit 6324f38

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

eval_protocol/cli.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -200,7 +200,7 @@ def _configure_parser(parser: argparse.ArgumentParser) -> argparse.ArgumentParse
200200
"training_config.gradient_accumulation_steps": "The number of batches to accumulate gradients before updating the model parameters. The effective batch size will be batch-size multiplied by this value.",
201201
"training_config.learning_rate_warmup_steps": "The number of learning rate warmup steps for the reinforcement fine-tuning job.",
202202
"mcp_server": "The MCP server resource name to use for the reinforcement fine-tuning job. (Optional)",
203-
"loss_config.method": "RL loss method for underlying trainers. One of {grpo,dapo}.",
203+
"loss_config.method": "RL loss method for underlying trainers. One of {grpo,dapo,gspo-token}.",
204204
}
205205

206206
create_rft_job_fn = Fireworks().reinforcement_fine_tuning_jobs.create

0 commit comments

Comments
 (0)