From 594b52b04ee1823cb108d785367a81ccb0175a46 Mon Sep 17 00:00:00 2001
From: "S.P. Mohanty" <spmohanty91@gmail.com>
Date: Tue, 19 Mar 2024 15:54:51 +0000
Subject: [PATCH] Align functionality of log_every_n_steps with documentation.
---
local_evaluation.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/local_evaluation.py b/local_evaluation.py
index 6ab525a..9e970e5 100644
--- a/local_evaluation.py
+++ b/local_evaluation.py
@@ -108,9 +108,9 @@ def evaluate_outputs(data_df, outputs, log_every_n_steps=1):
per_task_metrics[task_name]["sample_score"].append(metric_score)
- if row_idx % log_every_n_steps == 0:
+ if (row_idx + 1) % log_every_n_steps == 0:
print_sample(
- row_idx, model_output, ground_truth, metric, metric_score
+ row_idx + 1, model_output, ground_truth, metric, metric_score
)
return per_task_metrics
--
GitLab