diff --git a/src/decentralizepy/training/Training.py b/src/decentralizepy/training/Training.py index dbb636ce9ed87276ac6aadbdbab9b74395be4422..6696f357bbd631151be5a5f6a9eaa747a8b01493 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))