chunk_content: d):
      with torch.no_grad():
          logits, labels = eval_pred
          logits_tensor = torch.from_numpy(logits)
          logits_tensor = nn.functional.interpolate(
              logits_tensor,
              size=labels.shape[-2:],
              mode="bilinear",
              align_corners=False,
          ).argmax(dim=1)
  
          pred_labels = logits_tensor.detach().cpu().numpy()
          # currently using _compute instead of compute
          # see this i