From 3bae8f62fd5879b8448c0edb0758e2483b09b197 Mon Sep 17 00:00:00 2001 From: Marcin Chrzanowski Date: Sun, 2 May 2021 20:35:24 +0200 Subject: Remove prints --- src/counting_small.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/src/counting_small.py b/src/counting_small.py index 48d967c..06cd690 100644 --- a/src/counting_small.py +++ b/src/counting_small.py @@ -13,8 +13,6 @@ def loss_function(output, target): return torch.sum(output * errors ** 2) / output.shape[0] def count_correct(output, target): - print('output shape', output.shape) - print('target shape', target.shape) output = output.reshape(output.shape[0], 6, 10) target = target.reshape(target.shape[0], 6, 10) target = torch.argmax(target, dim=2) -- cgit v1.2.3