what is feat_dim in LGMLOSS_v0,How can I set it

YirongMao / softmax_variants

PyTorch code for softmax variants: center loss, cosface loss, large-margin gaussian mixture, COCOLoss, ring loss

252 stars 49 forks source link

Open AnnaRose0226 opened 5 years ago

AnnaRose0226 commented 5 years ago

what is feat_dim in LGMLOSS_v0,How can I set it

YirongMao commented 5 years ago

what is feat_dim in LGMLOSS_v0,How can I set it

Thanks for your interest. The feat_dim is the dimension of the input feature. If images are embedded into 2-D space, the feat_dim is set to be 2.

liuge1008 commented 3 years ago

@YirongMao 您好，请问如果是二分类的话，feat_dim设2，还需要加self.ip2这层吗，是不是只用一个全连接层就可以呢