diff options
| author | Marcin Chrzanowski <m@m-chrzan.xyz> | 2021-05-02 20:35:24 +0200 |
|---|---|---|
| committer | Marcin Chrzanowski <m@m-chrzan.xyz> | 2021-05-02 20:35:24 +0200 |
| commit | 3bae8f62fd5879b8448c0edb0758e2483b09b197 (patch) | |
| tree | 9e76f5db7f8e4dfe34163f27e4e943c3790e1dda | |
| parent | 194a2ac67bb956e05e5a076ecf6893a73e569589 (diff) | |
Remove prints
| -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) |