You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
CHECK_EQ(outer_num_ * inner_num_, bottom[1]->count())
<< "Number of labels must match number of predictions; " << "e.g., if softmax axis == 1and prediction shape is (N, C, H, W), "
<< "label count (number of labels) must be N*H*W, "
<< "with integer values in {0, 1, ..., C-1}.";
So the documentation should have been:
* -#
The text was updated successfully, but these errors were encountered:
Issue summary
The documentation for
SoftmaxWithLoss
is* -#
But here in the code this is what is checked:
So the documentation should have been:
* -#
The text was updated successfully, but these errors were encountered: