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.

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

[Training/Python] Add option to enable symbolic shape inference #5107

Open
wants to merge 13 commits into
base: master
from

Conversation

@KeDengMS
Copy link
Contributor

KeDengMS commented Sep 10, 2020

and allow save model to directory

Description: Describe your changes.

Motivation and Context

  • Why is this change required? What problem does it solve?
  • If it fixes an open issue, please link to the issue here.
…save model to directory
@KeDengMS KeDengMS requested review from BowenBao, liqunfu, spandantiwari, thiagocrepaldi and microsoft/onnxruntime as code owners Sep 10, 2020
KeDengMS added 7 commits Sep 10, 2020
@@ -198,23 +202,16 @@ def _preprocess(self, in_mp):

# returns True if no more ready nodes
def _insert_ready_nodes():
ready_nodes = [pn for pn in pending_nodes if all([i in defined for i in pn.input if i])]
ready_nodes = [pn for pn in pending_nodes if len(pn.input) == 0 or all([i in defined for i in pn.input if i])]

This comment has been minimized.

@RandySheriffH

RandySheriffH Sep 11, 2020

Contributor

len(pn.input) == 0 [](start = 57, length = 19)

nit:

if not pn.input #Resolved

This comment has been minimized.

@KeDengMS

KeDengMS Sep 12, 2020

Author Contributor

Thanks, rewritten this part


In reply to: 487200955 [](ancestors = 487200955)

KeDengMS added 5 commits Sep 11, 2020
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Linked issues

Successfully merging this pull request may close these issues.

None yet

2 participants
You can’t perform that action at this time.