VIT(Vision Transformer)学习(三)-纯VIT之swin transformer模型理解
classBalanceLoss(nn.Module):def__init__(self,ignore_index=255,reduction='mean',weight=None):super(BalanceLoss,self).__init__()self.ignore_label=ignore_indexself.reduction=reductionself.criterion=nn.NL