fix: accuracy as torchmetric fixed

examples/gmlvq_iris.py

@@ -97,6 +97,13 @@ def main():
         step=Steps.VALIDATION,
     )
 
+    accuracy = LogTorchmetricCallback(
+        'validation_accuracy',
+        torchmetrics.Accuracy,
+        num_classes=3,
+        step=Steps.VALIDATION,
+    )
+
     es = EarlyStopping(
         monitor=stopping_criterion.name,
         mode="max",
@@ -111,6 +118,7 @@ def main():
         callbacks=[
             vis,
             recall,
+            accuracy,
             stopping_criterion,
             es,
             PlotLambdaMatrixToTensorboard(),

prototorch/models/architectures/base.py

@@ -237,7 +237,7 @@ class BaseYArchitecture(pl.LightningModule):
         _, y = batch
         for metric in self.registered_metrics[step]:
             instance = self.registered_metrics[step][metric].to(self.device)
-            instance(y, preds)
+            instance(y, preds.reshape(y.shape))
 
     def update_metrics_epoch(self, step):
         for metric in self.registered_metrics[step]: