ldam专题

长尾问题之LDAM

做法&代码&公式 step1: 全连接层的权重W和特征向量X都归一化,相乘 W * X = P (得到各个类别的概率) # 定义权重,初始化weight = nn.Parameter(torch.FloatTensor(num_classes, num_features))weight.data.uniform_(-1, 1).renorm_(2, 1, 1e-5).mul_(1e5)#