mask_arg = (maxpred != self.ignore_index)

[HaoL20]

In the the forward() of classIW_MaxSquareloss, the code maxpred, argpred = torch.max(prob, 1) will return the values and indices, which means the maxpred is the max values and the argpred is the index.
The next line mask_arg = (maxpred != self.ignore_index), it`s going to select the unignore_index with the mask. Should it be changed like this:mask_arg = (argpred != self.ignore_index)？