remove reset

torch-points3d · Jul 28, 2021 · 11e8a5e · 11e8a5e
1 parent 413a2bd
commit 11e8a5e
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 8 deletions.
diff --git a/torch_points3d/metrics/base_tracker.py b/torch_points3d/metrics/base_tracker.py
@@ -57,8 +57,3 @@ def get_final_loss_metrics(self):
             metrics[key] = m.compute()
         self.loss_metrics = nn.ModuleDict()
         return metrics
-
-    def reset(self, stage: str = "train"):
-        self._finalised = False
-        self.stage = stage
-        self.loss_metrics = nn.ModuleDict()
diff --git a/torch_points3d/metrics/segmentation/segmentation_tracker.py b/torch_points3d/metrics/segmentation/segmentation_tracker.py
@@ -49,6 +49,3 @@ def _finalise(self):
         matrix = self.confusion_matrix_metric.compute()
         segmentation_metrics = self.compute_metrics_from_cm(matrix)
         return segmentation_metrics
-
-    def reset(self, stage: str = "train"):
-        super().reset(stage)