From 50106c68851d79c1ac03d49fb73936ad90dbbc8d Mon Sep 17 00:00:00 2001 From: hglee98 Date: Fri, 20 Dec 2024 06:53:51 +0000 Subject: [PATCH] [revert] delete predictions from evaluation_summary --- src/netspresso_trainer/pipelines/evaluation.py | 5 ++--- src/netspresso_trainer/utils/record.py | 1 - 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/src/netspresso_trainer/pipelines/evaluation.py b/src/netspresso_trainer/pipelines/evaluation.py index f2da5f64..cd91694d 100644 --- 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 index cd877c92..e7e54480 100644 --- 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