From f3207316389337fea8a3a2a78a5c8cad4ef9a1c7 Mon Sep 17 00:00:00 2001
From: Jeffrey Wigger <jeffrey.wigger@epfl.ch>
Date: Sun, 22 May 2022 16:13:33 +0200
Subject: [PATCH] fixed reddit debug
---
eval/run_xtimes_reddit_rw.sh | 2 +-
src/decentralizepy/datasets/Reddit.py | 4 ++--
2 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/eval/run_xtimes_reddit_rw.sh b/eval/run_xtimes_reddit_rw.sh
index fe90dcf..f32b877 100755
--- a/eval/run_xtimes_reddit_rw.sh
+++ b/eval/run_xtimes_reddit_rw.sh
@@ -43,7 +43,7 @@ procs_per_machine=16
machines=6
global_epochs=80
eval_file=testing.py
-log_level=INFO
+log_level=DEBUG
ip_machines=$nfs_home/configs/ip_addr_6Machines.json
diff --git a/src/decentralizepy/datasets/Reddit.py b/src/decentralizepy/datasets/Reddit.py
index 4bc3e2f..b0f91c1 100644
--- a/src/decentralizepy/datasets/Reddit.py
+++ b/src/decentralizepy/datasets/Reddit.py
@@ -484,7 +484,7 @@ class Reddit(Dataset):
count += 1
_, predictions = torch.max(outputs, 1)
for label, prediction in zip(labels, predictions):
- logging.debug("{} predicted as {}".format(label, prediction))
+ # logging.debug("{} predicted as {}".format(label, prediction))
if label == prediction:
correct_pred[label] += 1
total_correct += 1
@@ -498,7 +498,7 @@ class Reddit(Dataset):
accuracy = 100 * float(value) / total_pred[key]
else:
accuracy = 100.0
- logging.debug("Accuracy for class {} is: {:.1f} %".format(key, accuracy))
+ # logging.debug("Accuracy for class {} is: {:.1f} %".format(key, accuracy))
accuracy = 100 * float(total_correct) / total_predicted
loss_val = loss_val / count
--
GitLab