onnxruntime
Resolve issue with wrapped ORTModule load_state_dict
#7847

Merged

Commits

Encapsulate children modules inside a ModuleAccessor object to prevent erroneuos iteration over children while loading the state dictionary

baijumeswani committed 5 years ago
Add named_models, models, apply methods, change ModuleAccessor to ModuleMetadata and modify unit tests

baijumeswani committed 5 years ago
Change ModuleMetadata module getter logic, raise NotImplementedError for add_modules

baijumeswani committed 5 years ago
Add comment explaining why overriding _load_from_state_dict method is needed

baijumeswani committed 5 years ago