From 713c0ff3a8e37a1f9cadcf0ffc9c8acde88054d8 Mon Sep 17 00:00:00 2001
From: Rishi Sharma <rishi.sharma@epfl.ch>
Date: Thu, 27 Jan 2022 16:02:18 +0100
Subject: [PATCH] Removed extra logging
---
src/decentralizepy/training/Training.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/decentralizepy/training/Training.py b/src/decentralizepy/training/Training.py
index dbb636c..6696f35 100644
--- a/src/decentralizepy/training/Training.py
+++ b/src/decentralizepy/training/Training.py
@@ -124,12 +124,12 @@ class Training:
epoch_loss = 0.0
count = 0
for data, target in trainset:
- logging.info(
+ logging.debug(
"Starting minibatch {} with num_samples: {}".format(
count, len(data)
)
)
- logging.info("Classes: {}".format(target))
+ logging.debug("Classes: {}".format(target))
epoch_loss += self.trainstep(data, target)
count += 1
logging.info("Epoch: {} loss: {}".format(epoch, epoch_loss / count))
--
GitLab