Function batch_to_gpu

demo/Tacotron2/waveglow/data_function.py:73–78

(batch)

def batch_to_gpu(batch):
    x, y, len_y = batch
    x = to_gpu(x).float()
    y = to_gpu(y).float()
    len_y = to_gpu(torch.sum(len_y))
    return ((x, y), y, len_y)