INNER CODE UNIT · Python

masked_loss_matrix

yxuansu/SimCTG · dialogue_generation/loss_func.py:73

    masked_loss_matrix = loss_matrix * loss_mask

    loss_matrix = torch.sum(masked_loss_matrix, dim = -1)
    assert loss_matrix.size() == input_ids.size()
    loss_matrix = loss_matrix * input_mask
    cl_loss = torch.sum(loss_matrix) / torch.sum(loss_mask)
    return cl_loss
    

View source record →