import torch


def evaluator_loss(pred_iou, gt_iou):
    """
    MSE loss for evaluator

    pred_iou: [bs,]
    gt_iou: [bs,]
    """
    loss = torch.mean(0.5 * (pred_iou - gt_iou)**2)
    return loss