def init_weights(net, init_type='normal'):
    print(('initialization method [%s]' % init_type))
    if (init_type == 'normal'):
        net.apply(weights_init_normal)
    elif (init_type == 'xavier'):
        net.apply(weights_init_xavier)
    elif (init_type == 'kaiming'):
        net.apply(weights_init_kaiming)
    else:
        raise NotImplementedError(('initialization method [%s] is not implemented' % init_type))
def init_weights(net, init_type='normal'):
    print(('initialization method [%s]' % init_type))
    if (init_type == 'normal'):
        net.apply(weights_init_normal)
    elif (init_type == 'xavier'):
        net.apply(weights_init_xavier)
    elif (init_type == 'kaiming'):
        net.apply(weights_init_kaiming)
    else:
        raise NotImplementedError(('initialization method [%s] is not implemented' % init_type))