4326582bb1e68480ef79a02abbf4bfacc3aadede,datasets/TTSDataset.py,MyDataset,collate_fn,#MyDataset#Any#,173

Before Change

            item_idxs = [d["item_idx"] for d in batch]
            text = [d["text"] for d in batch]

            text_lenghts = np.array([len(x) for x in text])
            max_text_len = np.max(text_lenghts)

            // if specs are not computed, compute them.
            if batch[0]["mel"] is None and batch[0]["linear"] is None:

After Change

        if isinstance(batch[0], collections.Mapping):

            text_lenghts = np.array([len(d["text"]) for d in batch])
            text_lenghts, ids_sorted_decreasing = torch.sort(
                torch.LongTensor(text_lenghts), dim=0, descending=True)

            wav = [batch[idx]["wav"] for idx in ids_sorted_decreasing]
            item_idxs = [
                batch[idx]["item_idx"] for idx in ids_sorted_decreasing

Instances