Feat: Update nat

Feat: Update loader.py
speedcell4 · Sep 10, 2024 · 8900975 · 8900975
1 parent 4bf48f3
commit 8900975
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/torchglyph/metric.py b/torchglyph/metric.py
@@ -43,15 +43,11 @@ def update(self, sequence: Z) -> None:
 class HashMetric(MetricCollection):
     def __init__(self) -> None:
         super(HashMetric, self).__init__({
-            'loss': MeanMetric(),
             'pos': MeanMetric(),
             'neg': MeanMetric(),
             'unique': MeanMetric(),
         })
 
-    def update_loss(self, loss: Tensor) -> None:
-        self['loss'].update(loss)
-
     def update(self, x: Tensor, y: Tensor, t1: Tensor, t2: Tensor) -> None:
         pos = t1[:, None] == t2[None, :]
         neg = t1[:, None] != t2[None, :]