nobuco icon indicating copy to clipboard operation
nobuco copied to clipboard

Preserve node naming/namespacing

Open matemijolovic opened this issue 9 months ago • 4 comments

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!

matemijolovic avatar May 20 '24 12:05 matemijolovic