Try to extract layer names from Chain, Parallel and SkipConnection #95
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Fixes #94
This adds functionality to extract layer names from Chain, Parallel and SkipConnection.
This functionality is enabled by default, meaning that if the model to be serialized is a Chain with named layers, ONNXNaiveNASflux will use those names instead of the default ones. The name hunting will recurse to any nested Chains and into Parallels and SkipConnections.
@alicewith: You can see some examples of what this does in the tests, starting here.
Note that the input node is a special node in ONNX which does not exist in a Flux Chain, so just naming the first layer "input" will not give you the result you want; You have to give the name as input to
save
just as on current master.