Skip to content

[BUG] ZenFlow Stage 3 with full_warm_up_rounds=0 fails due to missing complete_column_offset attribute #7796

@Antlera

Description

@Antlera

Describe the bug
When running Stage 3 with full_warm_up_rounds=0, the training fails with the following error:
AttributeError: 'Parameter' object has no attribute 'complete_column_offset'

This seems to be related to attribute propagation or initialization during parameter setup.
Thanks @tohtana for pointing this out in the PR #7793.


To Reproduce
Steps to reproduce the behavior:

  1. Configure Stage 3 with full_warm_up_rounds=0
  2. Launch training as usual
  3. Observe the AttributeError during backward or parameter initialization

Expected behavior
Training should run normally without attribute errors, even when full_warm_up_rounds=0.


Additional context
This likely needs a small fix in attribute initialization or propagation logic.
I’ll open a follow-up PR once the root cause is confirmed.

Metadata

Metadata

Assignees

Labels

bugSomething isn't workingtraining

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions