something wrong in "weighted_softmax_loss_layer.cpp"

[yxliu-ntu]

          for (int k = 0; k < dim; ++k) {
            bottom_diff[i * dim + k * inner_num_ + j] *= w;
          }

should be changed to

          for (int k = 0; k < channels; ++k) {
            bottom_diff[i * dim + k * inner_num_ + j] *= w;
          }

[JayYangSS]

@Mithril-NTU
I think it should be:

for (int k = 0; k < bottom[0]->shape(softmax_axis_); ++k) {
            bottom_diff[i * dim + k * inner_num_ + j] *= w;
          }

otherwise, there will be error: channels: undecleared identifier

[zhanglaplace]

softmax_axis_ default is 1, So the bottom[0]->shape(softmax_axis_) == channels_ 👍