Skip to content

Commit 11e8a5e

Browse files
author
humanpose1
committed
remove reset
1 parent 413a2bd commit 11e8a5e

File tree

2 files changed

+0
-8
lines changed

2 files changed

+0
-8
lines changed

torch_points3d/metrics/base_tracker.py

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -57,8 +57,3 @@ def get_final_loss_metrics(self):
5757
metrics[key] = m.compute()
5858
self.loss_metrics = nn.ModuleDict()
5959
return metrics
60-
61-
def reset(self, stage: str = "train"):
62-
self._finalised = False
63-
self.stage = stage
64-
self.loss_metrics = nn.ModuleDict()

torch_points3d/metrics/segmentation/segmentation_tracker.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,3 @@ def _finalise(self):
4949
matrix = self.confusion_matrix_metric.compute()
5050
segmentation_metrics = self.compute_metrics_from_cm(matrix)
5151
return segmentation_metrics
52-
53-
def reset(self, stage: str = "train"):
54-
super().reset(stage)

0 commit comments

Comments
 (0)