We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 413a2bd commit 11e8a5eCopy full SHA for 11e8a5e
torch_points3d/metrics/base_tracker.py
@@ -57,8 +57,3 @@ def get_final_loss_metrics(self):
57
metrics[key] = m.compute()
58
self.loss_metrics = nn.ModuleDict()
59
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
@@ -49,6 +49,3 @@ def _finalise(self):
49
matrix = self.confusion_matrix_metric.compute()
50
segmentation_metrics = self.compute_metrics_from_cm(matrix)
51
return segmentation_metrics
52
53
54
- super().reset(stage)
0 commit comments