Fix accuracy calculation in batch_eth_mnist.py

BindsNET · Mar 22, 2024 · d9fed4a · d9fed4a
1 parent 775abcf
commit d9fed4a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/examples/mnist/batch_eth_mnist.py b/examples/mnist/batch_eth_mnist.py
@@ -363,7 +363,9 @@
     )
 
     # Compute network accuracy according to available classification strategies.
-    accuracy["all"] += float(torch.sum(label_tensor.long() == all_activity_pred.to(device)).item())
+    accuracy["all"] += float(
+        torch.sum(label_tensor.long() == all_activity_pred.to(device)).item()
+    )
     accuracy["proportion"] += float(
         torch.sum(label_tensor.long() == proportion_pred.to(device)).item()
     )