2a094e6fb2c8563e72f5b5a703951e5edcc673f3,train.py,,,#,444

Before Change

    // TODO: better way
    lengths = []
    for i in tqdm(range(len(X))):
        lengths.append(X[i].shape[0])
    lengths = np.array(lengths)

    // Prepare sampler
    sampler = PartialyRandomizedSimilarTimeLengthSampler(

After Change

    X = FileSourceDataset(RawAudioDataSource(data_root, speaker_id))
    if local_conditioning:
        Mel = FileSourceDataset(MelSpecDataSource(data_root, speaker_id))
        assert len(X) == len(Mel)
    else:
        Mel = None
    print(len(X))

Instances