64bba81a6867ad72aad461e27221ea278d25f5c1,models/networks.py,,init_weights,#,60

Before Change

    elif init_type == "xavier":
        net.apply(weights_init_xavier)
    elif init_type == "kaiming":
        net.apply(weights_init_kaiming)
    elif init_type == "orthogonal":
        net.apply(weights_init_orthogonal)
    else:
        raise NotImplementedError("initialization method [%s] is not implemented" % init_type)

After Change

        elif classname.find("BatchNorm2d") != -1:
            init.normal(m.weight.data, 1.0, 0.02)
            init.constant(m.bias.data, 0.0)
    return init_func


def init_net(net, init_type="normal", gpu_ids=[]):
    print("initialization method [%s]" % init_type)

Instances