[TTS] Fix TTS recipes with PTL 2.0 (#7188)

Signed-off-by: Ryan <[email protected]>
NVIDIA · Aug 9, 2023 · ada4fe5 · ada4fe5
1 parent 5cffd9a
commit ada4fe5
Show file tree

Hide file tree

Showing 7 changed files with 7 additions and 7 deletions.
diff --git a/examples/tts/conf/audio_codec/encodec.yaml b/examples/tts/conf/audio_codec/encodec.yaml
@@ -138,7 +138,7 @@ trainer:
   num_nodes: 1
   devices: 1
   accelerator: gpu
-  strategy: ddp
+  strategy: ddp_find_unused_parameters_true
   precision: 32 # Vector quantization only works with 32-bit precision.
   max_epochs: ${max_epochs}
   accumulate_grad_batches: 1

diff --git a/examples/tts/conf/hifigan/hifigan.yaml b/examples/tts/conf/hifigan/hifigan.yaml
@@ -72,7 +72,7 @@ trainer:
   num_nodes: 1
   devices: 1
   accelerator: gpu
-  strategy: ddp
+  strategy: ddp_find_unused_parameters_true
   precision: 32
   max_steps: ${model.max_steps}
   accumulate_grad_batches: 1

diff --git a/examples/tts/conf/hifigan/hifigan_44100.yaml b/examples/tts/conf/hifigan/hifigan_44100.yaml
@@ -72,7 +72,7 @@ trainer:
   num_nodes: 1
   devices: -1
   accelerator: gpu
-  strategy: ddp
+  strategy: ddp_find_unused_parameters_true
   precision: 16
   max_steps: ${model.max_steps}
   accumulate_grad_batches: 1

diff --git a/examples/tts/conf/hifigan_dataset/hifigan_22050.yaml b/examples/tts/conf/hifigan_dataset/hifigan_22050.yaml
@@ -129,7 +129,7 @@ trainer:
   num_nodes: 1
   devices: 1
   accelerator: gpu
-  strategy: ddp
+  strategy: ddp_find_unused_parameters_true
   precision: 16
   max_epochs: ${max_epochs}
   accumulate_grad_batches: 1

diff --git a/examples/tts/conf/hifigan_dataset/hifigan_44100.yaml b/examples/tts/conf/hifigan_dataset/hifigan_44100.yaml
@@ -129,7 +129,7 @@ trainer:
   num_nodes: 1
   devices: 1
   accelerator: gpu
-  strategy: ddp
+  strategy: ddp_find_unused_parameters_true
   precision: 16
   max_epochs: ${max_epochs}
   accumulate_grad_batches: 1

diff --git a/examples/tts/conf/univnet/univnet.yaml b/examples/tts/conf/univnet/univnet.yaml
@@ -78,7 +78,7 @@ trainer:
   num_nodes: 1
   devices: 1
   accelerator: gpu
-  strategy: ddp
+  strategy: ddp_find_unused_parameters_true
   precision: 32
   max_steps: ${model.max_steps}
   accumulate_grad_batches: 1

diff --git a/nemo/collections/tts/models/audio_codec.py b/nemo/collections/tts/models/audio_codec.py
@@ -267,7 +267,7 @@ def training_step(self, batch, batch_idx):
         self.log_dict(metrics, on_step=True, sync_dist=True)
         self.log("t_loss", train_loss_mel, prog_bar=True, logger=False, sync_dist=True)
 
-    def training_epoch_end(self, outputs):
+    def on_train_epoch_end(self):
         self.update_lr("epoch")
 
     def validation_step(self, batch, batch_idx):