Skip to content

[Feature] Add support for in-flight weight updates #496

@lewtun

Description

@lewtun

In-flight weight updates were proposed by PipelineRL and allow one to stay near on-policy without tanking throughput:

Image

It was also validated in the ScaleRL paper, that adopting in-flight weight updates dramatically speeds up training by factors of 2-3x over conventional off-by-k training:

Image

I poked around the torchforge codebase and did not see this implemented yet, but think it would be a high impact feature for the open source community to include :)

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions