Merge pull request #365 from jaeyoo/master

Fix TFLite notebooks for Tacotron2 and FastSpeech

Author: dathudeptrai

notebooks/TensorFlowTTS_FastSpeech_with_TFLite.ipynb

@@ -200,7 +200,7 @@
     "# initialize melgan model\n",
     "with open('../examples/melgan/conf/melgan.v1.yaml') as f:\n",
     "    melgan_config = yaml.load(f, Loader=yaml.Loader)\n",
-    "melgan_config = MelGANGeneratorConfig(**melgan_config[\"generator_params\"])\n",
+    "melgan_config = MelGANGeneratorConfig(**melgan_config[\"melgan_generator_params\"])\n",
     "melgan = TFMelGANGenerator(config=melgan_config, name='melgan_generator')\n",
     "melgan._build()\n",
     "melgan.load_weights(\"../examples/melgan/checkpoints/generator-1500000.h5\")"

notebooks/TensorFlowTTS_Tacotron2_with_TFLite.ipynb

@@ -197,7 +197,7 @@
     "# initialize melgan model\n",
     "with open('../examples/melgan/conf/melgan.v1.yaml') as f:\n",
     "    melgan_config = yaml.load(f, Loader=yaml.Loader)\n",
-    "melgan_config = MelGANGeneratorConfig(**melgan_config[\"generator_params\"])\n",
+    "melgan_config = MelGANGeneratorConfig(**melgan_config[\"melgan_generator_params\"])\n",
     "melgan = TFMelGANGenerator(config=melgan_config, name='melgan_generator')\n",
     "melgan._build()\n",
     "melgan.load_weights(\"../examples/melgan/checkpoints/generator-1500000.h5\")"