fixed bug in doc

Author: Johanmkr

CollaborativeCoding/metrics/precision.py

@@ -10,8 +10,8 @@ class Precision(nn.Module):
     ----------
     num_classes : int
         Number of classes in the dataset.
-    micro_averaging : bool
-        Performs micro-averaging if True, otherwise macro-averaging.
+    macro_averaging : bool
+        Performs macro-averaging if True, otherwise micro-averaging.
     """
 
     def __init__(self, num_classes: int, macro_averaging: bool = False):

main.py

@@ -145,7 +145,7 @@ def main():
         for x, y in tqdm(trainloader, desc="Training"):
             x, y = x.to(device), y.to(device)
             logits = model.forward(x)
-            loss = criterion(logits, y)
+            loss = criterion(logits, y) 
             loss.backward()
 
             optimizer.step()