Embed graph version in tensor names

2021-01-03 12:16:42 +00:00 · 2021-01-03 12:16:42 +00:00 · b7ce98552d
commit b7ce98552d
parent 21d2ea46b9
1 changed files with 1 additions and 1 deletions
--- a/training/deepspeech_training/train.py
+++ b/training/deepspeech_training/train.py
@ -98,7 +98,7 @@ class Dense(tf.keras.layers.Dense):

 class Model(tf.keras.Model):
    def __init__(self, dropout_rates=None, overlap=True, unroll=False, masking=True):
-        super().__init__()
+        super().__init__(name='DeepSpeech_v{}'.format(int(file_relative_read('GRAPH_VERSION').strip())))

        if dropout_rates is None:
            dropout_rates = defaultdict(lambda: 0.)