Fix #985

2021-11-24 12:48:16 +01:00 · 2021-11-24 12:48:16 +01:00 · 353f913efc
parent f1ea3ad182
commit 353f913efc
1 changed files with 1 additions and 0 deletions
--- a/TTS/vocoder/models/gan.py
+++ b/TTS/vocoder/models/gan.py
@ -325,6 +325,7 @@ class GAN(BaseVocoder):
            data_items (List): Data samples.
            verbose (bool): Log information if true.
            num_gpus (int): Number of GPUs in use.
+            rank (int): Rank of the current GPU. Defaults to None.

        Returns:
            DataLoader: Torch dataloader.