Improved state interoperability with complex 3rd-party nets
Added
- Flattened state dict export and loading via a
flatten
argument. This feature improves interoperability complex modules, that were not originally constructed with theco.Sequential
andco.Parallel
building blocks. - Context manager for triggering flattened state_dict export and loading.