Skip to content

Improved state interoperability with complex 3rd-party nets

Compare
Choose a tag to compare
@LukasHedegaard LukasHedegaard released this 23 Aug 08:53
· 375 commits to main since this release
25b624e

Added

  • Flattened state dict export and loading via a flatten argument. This feature improves interoperability complex modules, that were not originally constructed with the co.Sequential and co.Parallel building blocks.
  • Context manager for triggering flattened state_dict export and loading.