Skip to content

Fix TrainState field access and bump version to 0.4.1 - #22

Merged
JoKircher merged 1 commit into
mainfrom
fix/trainstate-field-names
Aug 3, 2026
Merged

Fix TrainState field access and bump version to 0.4.1#22
JoKircher merged 1 commit into
mainfrom
fix/trainstate-field-names

Conversation

@JoKircher

Copy link
Copy Markdown
Member

Access the Lux.Training.TrainState fields by their real names (parameters, states, optimizer_state) instead of the non-existent ps, st, opt_state, which would error at runtime. These field names are stable across the supported Lux range (>= 1.13).

Access the `Lux.Training.TrainState` fields by their real names
(`parameters`, `states`, `optimizer_state`) instead of the
non-existent `ps`, `st`, `opt_state`, which would error at runtime.
These field names are stable across the supported Lux range (>= 1.13).

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
@JoKircher
JoKircher requested a review from JulianTrommer August 3, 2026 08:51
@JoKircher
JoKircher merged commit 2a08fb2 into main Aug 3, 2026
1 check passed
@JulianTrommer
JulianTrommer deleted the fix/trainstate-field-names branch August 6, 2026 12:43
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.

2 participants