Metadata-Version: 2.1
Name: tacotron2-model
Version: 0.1.2
Summary: A PyPI port of the NVIDIA Tacotron2 model
Home-page: https://github.com/BenAAndrew/tacotron2-model
Author: Ben Andrew
Author-email: benandrew89@gmail.com
License: UNKNOWN
Description: # tacotron2-model
        A PyPI port of the NVIDIA Tacotron2 model
        
        Source: https://github.com/NVIDIA/tacotron2 (model.py)
        
        Only change from the NVIDIA original is a replacement of hparams with individual arguments to remove the dependency on tf.contrib.training.HParams (deprecated since tensorflow 1).
        
        ## Usage
        
        ```
        from tacotron2_model.tacotron2 import Tacotron2
        
        model = Tacotron2(N_MEL_CHANNELS, N_SYMBOLS, SYMBOLS_EMBEDDING_DIM, 
                          ENCODER_N_CONVOLUTIONS, ENCODER_EMBEDDING_DIM, 
                          ENCODER_KERNEL_SIZE, ATTENTION_RNN_DIM, ATTENTION_DIM,
                          ATTENTION_LOCATION_N_FILTERS, ATTENTION_LOCATION_KERNEL_SIZE,
                          DECODER_RNN_DIM, PRENET_DIM, MAX_DECODER_STEPS, GATE_THRESHOLD,
                          P_ATTENTION_DROPOUT, P_DECODER_DROPOUT, POSTNET_EMBEDDING_DIM,
                          POSTNET_KERNEL_SIZE, POSTNET_N_CONVOLUTIONS).cuda()
        
        print(model.eval())
        ```
        
        ## Example params
        
        ```
        N_MEL_CHANNELS = 80
        N_SYMBOLS = 148
        SYMBOLS_EMBEDDING_DIM = 512
        
        ENCODER_N_CONVOLUTIONS = 3
        ENCODER_EMBEDDING_DIM = 512
        ENCODER_KERNEL_SIZE = 5
        
        ATTENTION_RNN_DIM = 1024
        ATTENTION_DIM = 128
        ATTENTION_LOCATION_N_FILTERS = 32
        ATTENTION_LOCATION_KERNEL_SIZE = 31
        
        DECODER_RNN_DIM = 1024
        PRENET_DIM = 256
        MAX_DECODER_STEPS = 1000
        GATE_THRESHOLD = 0.5
        
        P_ATTENTION_DROPOUT = 0.1
        P_DECODER_DROPOUT = 0.1
        
        POSTNET_EMBEDDING_DIM = 512
        POSTNET_KERNEL_SIZE = 5
        POSTNET_N_CONVOLUTIONS = 5 
        ```
        
Platform: UNKNOWN
Classifier: Programming Language :: Python :: 3
Classifier: License :: OSI Approved :: BSD License
Classifier: Operating System :: OS Independent
Requires-Python: >=3.5
Description-Content-Type: text/markdown
