毛隆昆

Results 1 comments of 毛隆昆

`def cls_loss(pred, pred_aug, gold, pc_tran, aug_tran, pc_feat, aug_feat, ispn = True): mse_fn = torch.nn.MSELoss(reduce=True, size_average=True) cls_pc, cls_pc_raw= cal_loss_raw(pred, gold) cls_aug, cls_aug_raw= cal_loss_raw(pred_aug, gold) if ispn: cls_pc = cls_pc + 0.001*mat_loss(pc_tran)...