89ed262a88abf2721f9c74b6cdc9c885bf4e29e3,src/train_softmax.py,,get_symbol,#Any#Any#Any#,134

Before Change

  data_shape = (args.image_channel,args.image_h,args.image_w)
  image_shape = ",".join([str(x) for x in data_shape])
  if args.network[0]=="s":
    embedding = spherenet.get_symbol(512, args.num_layers)
  elif args.network[0]=="m":
    print("init mobilenet", args.num_layers)
    embedding = fmobilenet.get_symbol(512, 
        use_se=args.use_se, version_input=args.version_input, 

After Change

  data_shape = (args.image_channel,args.image_h,args.image_w)
  image_shape = ",".join([str(x) for x in data_shape])
  if args.network[0]=="d":
    embedding = fdensenet.get_symbol(512, args.num_layers,
        use_se=args.use_se, version_input=args.version_input, 
        version_output=args.version_output, version_unit=args.version_unit)
  elif args.network[0]=="m":
    print("init mobilenet", args.num_layers)
    embedding = fmobilenet.get_symbol(512, 
        use_se=args.use_se, version_input=args.version_input, 

Instances