Fix #985

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.