212abd8f1113e9eabf5248078acc54021d02268e,models/networks.py,,define_D,#Any#Any#Any#Any#Any#Any#Any#Any#,125

Before Change

    else:
        raise NotImplementedError("Discriminator model name [%s] is not recognized" %
                                  which_model_netD)
    if use_gpu:
        netD.cuda(gpu_ids[0])
    init_weights(netD, init_type=init_type)
    return netD

After Change

    if use_gpu:
        assert(torch.cuda.is_available())
        netD.cuda(gpu_ids[0])
        if len(gpu_ids) > 1:
            netD = torch.nn.DataParallel(netD, gpu_ids)
    init_weights(netD, init_type=init_type)
    return netD

Instances