        accelerator,
        epoch,
        completed_steps,
        train_loss,
    model.eval()

    eval_progress_bar = tqdm(dataloader, desc=f"Evaluating Epoch {epoch}", leave=False)
    for batch in eval_progress_bar:
        with torch.no_grad():
            # Forward pass
            outputs = model(**batch)
