I don’t know if you could be interested to explore the upstream diff on how they are refactoring the model (to remove also the old contrib):

  

      github.com/mozilla/DeepSpeech
  

  
    
  
    
  

  
    
      Low touch upgrade to TensorFlow 2.3 15
    

    
      mozilla:master ← mozilla:low-touch-r2.3
    

    
      
        opened 
        
          
        
        Jan 2, 2021
      
      

      
        
          
          reuben
        
      

      
        
          +476
          -1252
        
      
    
  


  
    Keeps changes to a minimum by leveraging the fact that under a `tfv1.Session` ob…ject, TensorFlow v1 style meta-graph construction works normally, including placeholders. This lets us keep changes to a minimum. The main change comes in the model definition code: the previous LSTMBlockCell/static_rnn/CudnnRNN parametrized RNN implementation gets replaced by `tf.keras.layers.LSTM` which is supposed to use the most appropriate implementation given the layer configuration and host machine setup. This is a graph breaking change and so GRAPH_VERSION is bumped.