mirror of https://github.com/coqui-ai/TTS.git
Fix #985
This commit is contained in:
parent
f1ea3ad182
commit
353f913efc
|
@ -325,6 +325,7 @@ class GAN(BaseVocoder):
|
||||||
data_items (List): Data samples.
|
data_items (List): Data samples.
|
||||||
verbose (bool): Log information if true.
|
verbose (bool): Log information if true.
|
||||||
num_gpus (int): Number of GPUs in use.
|
num_gpus (int): Number of GPUs in use.
|
||||||
|
rank (int): Rank of the current GPU. Defaults to None.
|
||||||
|
|
||||||
Returns:
|
Returns:
|
||||||
DataLoader: Torch dataloader.
|
DataLoader: Torch dataloader.
|
||||||
|
|
Loading…
Reference in New Issue