ttslearn.train_util.collate_fn_tacotron

ttslearn.train_util.collate_fn_tacotron(batch, reduction_factor=1)[source]

Collate function for Tacotron.

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

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

Returns

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

Return type

tuple