From 8505805dadecd12a7047574bc1970bcdb21440ab Mon Sep 17 00:00:00 2001
From: Yibing Liu <liuyibing01@baidu.com>
Date: Mon, 13 Apr 2020 11:31:39 +0000
Subject: [PATCH] Fix init in synthesis & remove dry run

---
 examples/waveflow/synthesis.py | 11 +----------
 examples/waveflow/waveflow.py  |  6 ------
 2 files changed, 1 insertion(+), 16 deletions(-)

diff --git a/examples/waveflow/synthesis.py b/examples/waveflow/synthesis.py
index 15c4d3b..b9569bf 100644
--- a/examples/waveflow/synthesis.py
+++ b/examples/waveflow/synthesis.py
@@ -93,16 +93,7 @@ def synthesize(config):
 
         # Build model.
         model = WaveFlow(config, checkpoint_dir)
-        model.build(training=False)
-        # Obtain the current iteration.
-        if config.checkpoint is None:
-            if config.iteration is None:
-                iteration = io.load_latest_checkpoint(checkpoint_dir)
-            else:
-                iteration = config.iteration
-        else:
-            iteration = int(config.checkpoint.split('/')[-1].split('-')[-1])
-
+        iteration = model.build(training=False)
         # Run model inference.
         model.infer(iteration)
 
diff --git a/examples/waveflow/waveflow.py b/examples/waveflow/waveflow.py
index 700116b..23c558e 100644
--- a/examples/waveflow/waveflow.py
+++ b/examples/waveflow/waveflow.py
@@ -81,12 +81,6 @@ class WaveFlow():
 
         waveflow = WaveFlowModule(config)
 
-        # Dry run once to create and initalize all necessary parameters.
-        audio = dg.to_variable(np.random.randn(1, 16000).astype(self.dtype))
-        mel = dg.to_variable(
-            np.random.randn(1, config.mel_bands, 63).astype(self.dtype))
-        waveflow(audio, mel)
-
         if training:
             optimizer = fluid.optimizer.AdamOptimizer(
                 learning_rate=config.learning_rate,
-- 
GitLab