Edit Tensorflow session


It is possible to edit Tensorflow session with DL Python Network Editor? I want to add ‘device_count={‘CPU’: NUM_PARALLEL_EXEC_UNITS}’ to Tensorflow ConfigProto. I saw that the session is set in the tensorflow_backend.py file and I’m wondering if is possible to modify it with DL Python Network Editor node. And in the future I want to set other parameters to the Tensorflow session in order to tune the performance.

Thank you,

Hi Mihai,

It is not possible to set the session in a “DL Python Network Editor” such that it is used in subsequent nodes. The model is saved as a “h5” file between the nodes which does not contain information about the Session.
You can only change the session used in the current Python node by calling keras.backend.set_session(sess).


1 Like

This topic was automatically closed 7 days after the last reply. New replies are no longer allowed.