onnxruntime/orttraining
baijumeswani ddf4aaaae1
Resolve issue with wrapped ORTModule load_state_dict (#7847)
* Encapsulate children modules inside a ModuleAccessor object to prevent erroneuos iteration over children while loading the state dictionary

* Add named_models, models, apply methods, change ModuleAccessor to ModuleMetadata and modify unit tests

* Change ModuleMetadata module getter logic, raise NotImplementedError for add_modules

* Add comment explaining why overriding _load_from_state_dict method is needed
2021-05-27 16:11:37 -07:00
..
orttraining Resolve issue with wrapped ORTModule load_state_dict (#7847) 2021-05-27 16:11:37 -07:00
pytorch_frontend_examples Sync ORTModule branch with master and fix tests (#6526) 2021-02-02 08:59:56 -08:00
tools Update BERT convergence baseline. 2021-05-25 17:11:46 -07:00