You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
The automatic checkpoint migration was temporarily rolled back due to legacy compatibility issues.
We still recommend you to use the upgrade guide and migrate completely to the Orbax API to ensure stability.
Or alternatively, add flax.config.update('flax_use_orbax_checkpointing', True) to your project to avoid being impacted by the automatic migration process.
reacted with thumbs up emoji reacted with thumbs down emoji reacted with laugh emoji reacted with hooray emoji reacted with confused emoji reacted with heart emoji reacted with rocket emoji reacted with eyes emoji
-
Changelog
flax.config.update('flax_use_orbax_checkpointing', True)
to your project to avoid being impacted by the automatic migration process.register_keypaths
.List of PRs
register_keypaths
by @copybara-service in Migrate Flax away fromregister_keypaths
#2963Full Changelog: v0.6.7...v0.6.8
This discussion was created from the release Version 0.6.8.
Beta Was this translation helpful? Give feedback.
All reactions