label-smooth, amsoftmax, partial-fc, focal-loss, triplet-loss, lovasz-softmax. Maybe useful
2.17k
stars
374
forks
source link
请问am-softmax可以用在多分类任务上么,比如网络结构最后一层输出分类是7,那amsoftmax()中的参数填啥,第一个in_feats应该填7,第二个num_class也填7,这应该不对,不知道怎么使用,请教了 #16
Closed
wbb95 closed 3 years ago
Hi,
An example is like this, you can have a try: