Function eval_step

scripts/release_notes/classifier.py:277–281

(batch, model, loss)

@torch.no_grad()
def eval_step(batch, model, loss):
    inpt, targets = batch
    output = model(inpt)
    l = loss(output, targets)
    return l


def balance_dataset(dataset: List):