Fix printing of Mean and Variance

tysam-code · Jan 31, 2023 · 8959293 · 8959293
1 parent e44a18d
commit 8959293
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/main.py b/main.py
@@ -639,4 +639,4 @@ def main():
     acc_list = []
     for run_num in range(25):
         acc_list.append(torch.tensor(main()))
-    print("Mean and variance:", (torch.mean(torch.stack(acc_list)), torch.var(torch.stack(acc_list)).item()).item())
+    print("Mean and variance:", (torch.mean(torch.stack(acc_list)).item(), torch.var(torch.stack(acc_list)).item()))