Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Resolve issue with wrapped ORTModule load_state_dict #7847

Merged
merged 4 commits into from
May 27, 2021
Merged

Resolve issue with wrapped ORTModule load_state_dict #7847

merged 4 commits into from
May 27, 2021

Commits on May 26, 2021

  1. Encapsulate children modules inside a ModuleAccessor object to preven… 

    …t erroneuos iteration over children while loading the state dictionary
    @baijumeswani
    baijumeswani committed May 26, 2021
    Configuration menu
    Copy the full SHA
    754f098 View commit details
    Browse the repository at this point in the history

Commits on May 27, 2021

  1. Add named_models, models, apply methods, change ModuleAccessor to Mod… 

    …uleMetadata and modify unit tests
    @baijumeswani
    baijumeswani committed May 27, 2021
    Configuration menu
    Copy the full SHA
    63f0707 View commit details
    Browse the repository at this point in the history

  2. Change ModuleMetadata module getter logic, raise NotImplementedError … 

    …for add_modules
    @baijumeswani
    baijumeswani committed May 27, 2021
    Configuration menu
    Copy the full SHA
    741f102 View commit details
    Browse the repository at this point in the history

  3. Add comment explaining why overriding _load_from_state_dict method is… 

    … needed
    @baijumeswani
    baijumeswani committed May 27, 2021
    Configuration menu
    Copy the full SHA
    3291659 View commit details
    Browse the repository at this point in the history