ttslearn.contrib.multispk_util.collate_fn_ms_tacotron

ttslearn.contrib.multispk_util.collate_fn_ms_tacotron(batch, reduction_factor=1)[source]

Collate function for multi-speaker Tacotron.

Parameters
  • batch (list) – List of tuples of the form (inputs, targets, spk_ids).

  • reduction_factor (int, optional) – Reduction factor. Defaults to 1.

Returns

Batch of inputs, input lengths, targets, target lengths, stop flags and spk ids.

Return type

tuple