0a8e27413d721bf8d753e5e6061cc24f5bf6474f,train.py,,,#,917

Before Change

    // Model
    model = build_model()
    print(model)
    if use_cuda:
        model = model.cuda()

    receptive_field = model.receptive_field

After Change

    // Dataloader setup
    data_loaders = get_data_loaders(data_root, speaker_id, test_shuffle=True)

    device = torch.device("cuda" if use_cuda else "cpu")

    // Model
    model = build_model().to(device)

Instances