nobuco
nobuco copied to clipboard
Preserve node naming/namespacing
Hello,
Thanks for developing this great tool! I have a question regarding layer/node naming in the converted Keras model.
For the context, I have a usecase where I postprocess the converted graph and I expect the nodes to conform to some custom naming schema (more specifically, to be "namespaced" or to contain some specific keywords). In the current Nobuco setup, module names in the Keras graph are autogenerated and thus not connected to Pytorch module names.
Do you think it would make sense to support some sort of naming/namespacing here? For 1-1 mappings PT module name could be used for Keras layer name, and for 1-N cases I'd propose using the PT module name as a prefix to the autogenerated name.
If you like this idea, I can offer some help with the implementation. Thanks in advance!