Skip to content

[skyrl-train][algorithm] Apply inference engine top-k mask to trainer #877

@erictang000

Description

@erictang000

As described in the Deepseek v3.2 Tech report - if using top_p or top_k sampling, the sampling mask should be propagated to the trainer to avoid drifting off policy in the trainer compared to the inference engine.

Image

Metadata

Metadata

Assignees

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions