m-chrzan.xyz
aboutsummaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/counting_small.py2
1 files changed, 0 insertions, 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)