Skip to content

Commit 303e5de

Browse files
committed
💕 ParallelWaveGAN HeNormal TF2.2 compatible, change default version to TF2.3.
1 parent be1cc38 commit 303e5de

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

‎setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@
2222
# TODO(@dathudeptrai) update requirement if needed.
2323
requirements = {
2424
"install": [
25-
"tensorflow-gpu>=2.2.0",
25+
"tensorflow-gpu>=2.3.0",
2626
"tensorflow-addons>=0.10.0",
2727
"setuptools>=38.5.1",
2828
"librosa>=0.7.0",

‎tensorflow_tts/models/parallel_wavegan.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ def get_initializer(initializer_seed=42):
2525
Returns:
2626
HeNormal initializer with seed = `initializer_seed`.
2727
"""
28-
return tf.keras.initializers.HeNormal(seed=initializer_seed)
28+
return tf.keras.initializers.he_normal(seed=initializer_seed)
2929

3030

3131
class TFConv1d1x1(tf.keras.layers.Conv1D):

0 commit comments

Comments
 (0)