diff --git a/eval/run_xtimes_reddit_rw.sh b/eval/run_xtimes_reddit_rw.sh index fe90dcfe39aefe2f1e402462d3d45be6a7c16503..f32b87708af8fac39066bff0e4326b63763dd5b0 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 4bc3e2f556b9e52fa254b90f98e141523ed94b2d..b0f91c10b3e96793374707efe4796800db4647de 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