flax
842a2874 - Set default types in Flax for Orbax restoration and add `restore_with_serialized_types` in preparation for an upcoming change.

Commit

2 years ago

Set default types in Flax for Orbax restoration and add `restore_with_serialized_types` in preparation for an upcoming change. PiperOrigin-RevId: 544069194

Author

cpgaffney1

Committer

a-googler

Parents

7c9c67f8

flax 842a2874 - Set default types in Flax for Orbax restoration and add `restore_with_serialized_types` in preparation for an upcoming change.

flax
842a2874 - Set default types in Flax for Orbax restoration and add `restore_with_serialized_types` in preparation for an upcoming change.