ttslearn.train_util.collate_fn_wavenet¶

ttslearn.train_util.collate_fn_wavenet(batch, max_time_frames=100, hop_size=80, aux_context_window=2)[source]¶

Collate function for WaveNet.

Parameters

batch (list) – List of tuples of the form (inputs, targets).
max_time_frames (int, optional) – Number of time frames. Defaults to 100.
hop_size (int, optional) – Hop size. Defaults to 80.
aux_context_window (int, optional) – Auxiliary context window. Defaults to 2.

Returns

Batch of waveforms and conditional features.

Return type