Resolve issue with wrapped ORTModule load_state_dict - #7847

Merged

Baiju Meswani (baijumeswani) merged 4 commits into

bmeswani/load-state-dict_wrapped-module

May 27, 2021

Commits on May 26, 2021

Encapsulate children modules inside a ModuleAccessor object to prevent erroneuos iteration over children while loading the state dictionary
Baiju Meswani (baijumeswani)
committed
754f098

Commits on May 27, 2021

Add named_models, models, apply methods, change ModuleAccessor to ModuleMetadata and modify unit tests
Baiju Meswani (baijumeswani)
committed
63f0707
Change ModuleMetadata module getter logic, raise NotImplementedError for add_modules
Baiju Meswani (baijumeswani)
committed
741f102
Add comment explaining why overriding _load_from_state_dict method is needed
Baiju Meswani (baijumeswani)
committed
3291659