-
Notifications
You must be signed in to change notification settings - Fork 4.3k
Fix issue exporting graph with multi-GPU #2573
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
Conversation
|
Updated this PR to remove changes to |
Our multi-GPU training had a regression such that freezing the graph was broken. This change fixes that issue by making a few changes: * Removes the top level "tower" variable scope added by multi-GPU so that the output nodes have correct names * Removes the use of "freeze_graph" and updates the tensorflow to barracuda converter to accept GraphDef objects * Adds the "auto reuse" to network layers which require them
c461a25
to
9d6f356
Compare
Our multi-GPU training had a regression such that freezing the
graph was broken. This change fixes that issue by making a few
changes:
so that the output nodes have correct names
barracuda converter to accept GraphDef objects