dme update

This commit is contained in:
erogol 2020-05-18 13:31:14 +02:00
parent df8fd3823d
commit 327c88b4bb
2 changed files with 13 additions and 7 deletions

View File

@ -1,12 +1,20 @@
## Utilities to Convert Models to Tensorflow2
Here there are utilities to convert trained Torch models to Tensorflow (2.2>=).
Here there are experimental utilities to convert trained Torch models to Tensorflow (2.2>=).
We currently support Tacotron2 with Location Sensitive Attention.
Converting Torch models to TF enables all the TF toolkit to be used for better deployment and device specific optimizations.
Be aware that our old Torch models may not work with this module due to additional changes in layer naming convention. Therefore, you need to train new models or handle these changes.
We do not plan to share training scripts for Tensorflow in near future. But any contribution in that direction would be more than welcome.
Note that we do not plan to share training scripts for Tensorflow in near future. But any contribution in that direction would be more than welcome.
To see how you can use TF model at inference, check the notebook.
This is an experimental release. If you encounter an error, please put an issue or in the best send a PR but you are mostly on your own.
### Converting a Model
- Run ```convert_tacotron2_torch_to_tf.py --torch_model_path /path/to/torch/model.pth.tar --config_path /path/to/model/config.json --output_path /path/to/output/tf/model``` with the right arguments.
### Known issues ans limitations
- We use a custom model load/save mechanism which enables us to store model related information with models weights. (Similar to Torch). However, it is prone to random errors.
- Current TF model implementation is slightly slower than Torch model. Hopefully, it'll get better with improving TF support for eager mode and ```tf.function```.
- TF implementation of Tacotron2 only supports regular Tacotron2 as in the paper.
- You can only convert models trained after TF model implementation since model layers has been updated in Torch model.

View File

@ -1,2 +0,0 @@
fuzzywuzzy
tensorflow>=2.2.0