Skip to content

Dataclamp checkpointing#786

Merged
kyralianaka merged 14 commits intomainfrom
dataclamp-checkpointing
Jan 27, 2026
Merged

Dataclamp checkpointing#786
kyralianaka merged 14 commits intomainfrom
dataclamp-checkpointing

Conversation

@kyralianaka
Copy link
Contributor

As described in issue #785, fixes checkpointing for networks with clamped variables of different sizes.

@kyralianaka
Copy link
Contributor Author

Also to note, clamp() also failed when using checkpointing + different numbers of cells per clamped state, so I added it to the test

@kyralianaka kyralianaka requested a review from jnsbck January 26, 2026 10:35
Copy link
Contributor

@jnsbck jnsbck left a comment

Choose a reason for hiding this comment

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

thanks, looks good! :)

@kyralianaka kyralianaka merged commit 322247c into main Jan 27, 2026
2 checks passed
@kyralianaka kyralianaka deleted the dataclamp-checkpointing branch January 27, 2026 14:22
@jnsbck jnsbck linked an issue Jan 28, 2026 that may be closed by this pull request
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.

Checkpointing issue with multiple data clamps

2 participants