Fix #985

2026-09-09 11:16:00 +08:00 · 2022-02-25 10:48:02 +01:00
parent f1ea3ad182
commit 353f913efc
1 changed files with 1 additions and 0 deletions
@@ -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.