diff options
Diffstat (limited to 'src/counting_small.py')
| -rw-r--r-- | src/counting_small.py | 2 |
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) |