Skip to content

Add Cosmos3-Super VideoPhy-2 Reasoner SFT recipe#101

Merged
lfengad merged 1 commit into
mainfrom
docs/finetune-cosmos3-super-reasoner
Jul 10, 2026
Merged

Add Cosmos3-Super VideoPhy-2 Reasoner SFT recipe#101
lfengad merged 1 commit into
mainfrom
docs/finetune-cosmos3-super-reasoner

Conversation

@Xuanmeng-Zhang

Copy link
Copy Markdown
Collaborator
  • Add Cosmos3-Super VideoPhy-2 Reasoner SFT recipe. It fully fine-tunes Qwen3-VL-32B using FSDP full sharding and initializes from the Cosmos3-Super language model merged with the 32B visual tower.

  • The new experiment adds a launcher and TOML recipe.

  • Use a learning rate of 1e-6 for both Nano and Super.

  • Set lr_multipliers={"model.visual": 1.0} to override the reasoner default of 0.1 and train the visual projector at the base learning rate. The previous mm_projector and merger keys matched no Qwen3-VL parameters because the projector is named model.visual.merger, so the intended 20x multiplier was never applied.

  • Document the Super recipe in docs/training.md and examples/README.md.

Add the Super-tier counterpart to the Nano VideoPhy-2 recipe. It fully fine-tunes Qwen3-VL-32B
    using FSDP full sharding and initializes from the Cosmos3-Super language model merged with the 32B
    visual tower.

The new experiment mirrors the Nano configuration but uses the 32B backbone and sets
    data_parallel_shard_degree=-1. It also adds a launcher and TOML recipe.

Use a learning rate of 1e-6 for both tiers. In a 150-iteration Super run, 1e-5 caused a temporary
    loss spike to about 6.6 at iteration 23. Both learning rates reached the same last-20 average loss of
    0.437, so 1e-6 provides the same result without the instability.

Set lr_multipliers={"model.visual": 1.0} to override the reasoner default of 0.1 and train the
    visual projector at the base learning rate. The previous mm_projector and merger keys matched no Qwen3-
    VL parameters because the projector is named model.visual.merger, so the intended 20x multiplier was
    never applied.

Document the Super recipe in docs/training.md and examples/README.md.
@Xuanmeng-Zhang Xuanmeng-Zhang requested a review from xlu451 July 10, 2026 05:24
@lfengad lfengad merged commit 3d9c087 into main Jul 10, 2026
8 checks passed
@lfengad lfengad deleted the docs/finetune-cosmos3-super-reasoner branch July 10, 2026 06:37
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants