5d9cfbd469ec9866c22c1d699f8eb08bdf8321fa,src/train_softmax.py,,get_symbol,#Any#Any#Any#,126

Before Change

    m = args.margin_m
    assert(s>0.0)
    assert(m>0.0)
    _weight = mx.symbol.Variable("fc7_weight", shape=(args.num_classes, args.emb_size), lr_mult=1.0)
    _weight = mx.symbol.L2Normalization(_weight, mode="instance")
    nembedding = mx.symbol.L2Normalization(embedding, mode="instance", name="fc1n")*s
    fc7 = mx.sym.FullyConnected(data=nembedding, weight = _weight, no_bias = True, num_hidden=args.num_classes, name="fc7")
    s_m = s*m

After Change

  elif args.network[0]=="s":
    print("init spherenet", args.num_layers)
    embedding = spherenet.get_symbol(args.emb_size, args.num_layers)
  elif args.network[0]=="y":
    print("init mobilefacenet", args.num_layers)
    embedding = fmobilefacenet.get_symbol(args.emb_size)
  else:
    print("init resnet", args.num_layers)
    embedding = fresnet.get_symbol(args.emb_size, args.num_layers, 
        version_se=args.version_se, version_input=args.version_input, 

Instances