dmlc / keras

Deep Learning library for Python. Convnets, recurrent neural networks, and more. Runs on MXNet, Theano or TensorFlow.
http://keras.io/
Other
125 stars 34 forks source link

mxnet_backend.clear_session should clear the placeholder name dictionary #71

Open Neiko2002 opened 7 years ago

Neiko2002 commented 7 years ago

Hello together,

right now I try to use the Keras (MXNet) library as a model converter. If you can port an existing MXNet model to a Keras model then you can convert from this intermediate format to the Theano, CNTK and TensorFlow format. This also works the other way around.

To make this happen I need access to the names of all nodes. Unfortunately the input variable name has an appended number which increases all the time. There is no way to reset it.

It would be helpful if mxnet_backend.clear_session() would reset the placeholder_name_dict variable.