Remove averaging in Accuracy

Author: amorehead

project/lit_image_classifier.py

@@ -38,9 +38,9 @@ def __init__(self, backbone, num_epochs: int = 5, lr=1e-4):
         self.loss_fn = nn.CrossEntropyLoss()
 
         # Define cross-validation metrics
-        self.train_acc = pl.metrics.Accuracy(average='weighted', num_classes=10)
-        self.val_acc = pl.metrics.Accuracy(average='weighted', num_classes=10)
-        self.test_acc = pl.metrics.Accuracy(average='weighted', num_classes=10)
+        self.train_acc = pl.metrics.Accuracy()
+        self.val_acc = pl.metrics.Accuracy()
+        self.test_acc = pl.metrics.Accuracy()
 
     def forward(self, x):
         # use forward for inference/predictions