In loss function: ```python neg_avg_factor = (1 - p_neg_weight).sum() ... cls_neg_loss = sum(cls_neg_loss_list) / neg_avg_factor ``` Why not use neg_avg_factor = (p_neg_weight).sum() ?