[revert] delete predictions from evaluation_summary

Nota-NetsPresso · Dec 20, 2024 · 50106c6 · 50106c6
1 parent c25d548
commit 50106c6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/src/netspresso_trainer/pipelines/evaluation.py b/src/netspresso_trainer/pipelines/evaluation.py
@@ -120,14 +120,13 @@ def log_end_evaluation(
             elapsed_time=time_for_evaluation,
         )
         predictions = self.task_processor.get_predictions(valid_samples, self.logger.class_map)
-        self.save_summary(losses, metrics, predictions, time_for_evaluation)
+        self.save_summary(losses, metrics, time_for_evaluation)
 
-    def save_summary(self, losses, metrics, predictions, time_for_evaluation):
+    def save_summary(self, losses, metrics, time_for_evaluation):
         flops, params = get_params_and_flops(self.model, self.sample_input.float())
         evaluation_summary = EvaluationSummary(
             losses=losses,
             metrics=metrics,
-            predictions=predictions,
             metrics_list=self.metric_factory.metric_names,
             primary_metric=self.metric_factory.primary_metric,
             flops=flops,

diff --git a/src/netspresso_trainer/utils/record.py b/src/netspresso_trainer/utils/record.py
@@ -161,7 +161,6 @@ class EvaluationSummary:
     primary_metric: str
     flops: Optional[int] = None
     params: Optional[int] = None
-    predictions: Optional[dict] = None
     total_evaluation_time: Optional[float] = None
     success: bool = False