diff options
| -rw-r--r-- | text_recognizer/models/base.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/text_recognizer/models/base.py b/text_recognizer/models/base.py index 64d3d71..90752e9 100644 --- a/text_recognizer/models/base.py +++ b/text_recognizer/models/base.py @@ -33,9 +33,9 @@ class LitBase(LightningModule): self.mapping = mapping # Placeholders - self.train_acc = Accuracy() - self.val_acc = Accuracy() - self.test_acc = Accuracy() + self.train_acc = Accuracy(mdmc_reduce="samplewise") + self.val_acc = Accuracy(mdmc_reduce="samplewise") + self.test_acc = Accuracy(mdmc_reduce="samplewise") def optimizer_zero_grad( self, |