Skip to content

refactor(args): rename --ema-rollout-policy to --rollout-weights and document the role flags - #265

Merged
Rockdu merged 3 commits into
mainfrom
fsdp-followups/1-docs-rename
Oct 4, 2026
Merged

Rockdu merged 3 commits into
mainfrom
fsdp-followups/1-docs-rename

Conversation

@Rockdu

@Rockdu Rockdu commented Sep 27, 2026 •

Copy link
Copy Markdown
Collaborator

Stack 1/9 of follow-ups to #263: main ← this ← #266 ← #267 ← #268 ← #269 ← #271 ← #272 ← #273 ← #274.

What

  • Rename --ema-rollout-policy {live,ema} to --rollout-weights {actor,ema} (supersedes refactor(args): rename --ema-rollout-policy to --rollout-weights #264).
  • CLI reference rows for --ref-load, --ref-lora-adapter-path, --ref-cpu-offload, --teacher-load, --teacher-lora-adapter-path, --teacher-cpu-offload, --lora-adapter-path and --ref-mode ref.
  • Drop a stale "no tensor snapshot manager" note from docs/fsdp_model_roles.md.

Why

The flag chooses which weights the rollout engines receive; "policy" and "live" did not say that.

Files

  • miles/utils/arguments.py, miles/backends/fsdp_utils/actor.py: the rename.
  • docs/user-guide/cli-reference.md, docs/fsdp_model_roles.md, recipes: docs and flag spelling.

Checklist

  • pre-commit run --all-files passes — not run locally
  • Added/updated tests for new behaviour — rename only, existing tests updated
  • pytest -x is green — CPU tests touched by the rename pass locally
  • If launch flags changed, python3 train.py --help still parses — not run (needs the GPU image)
  • If a public flag was added, it appears in the CLI reference docs
  • If an example was added, it has a real walkthrough — n/a

@zhihengy zhihengy left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@Rockdu
Rockdu merged commit cc22f10 into main Oct 4, 2026
58 of 63 checks passed
OnePunchMonk pushed a commit to OnePunchMonk/miles_diffusion that referenced this pull request Oct 5, 2026
…document the role flags (radixark#265)

* refactor(args): rename --ema-rollout-policy to --rollout-weights

* docs: list the reference, teacher and actor-adapter flags in the CLI reference

* docs(fsdp): drop a stale note about a removed snapshot manager
OnePunchMonk pushed a commit to OnePunchMonk/miles_diffusion that referenced this pull request Oct 5, 2026
OnePunchMonk pushed a commit to OnePunchMonk/miles_diffusion that referenced this pull request Oct 5, 2026
…document the role flags (radixark#265)

* refactor(args): rename --ema-rollout-policy to --rollout-weights

* docs: list the reference, teacher and actor-adapter flags in the CLI reference

* docs(fsdp): drop a stale note about a removed snapshot manager
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

run-ci-basic Authorize and run only basic unlabeled CI tests

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants