From 61b6934f914b1615483a255419ac8471853901e9 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Tue, 13 Sep 2022 18:13:30 +0200 Subject: Fix acc metric --- text_recognizer/models/base.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'text_recognizer') 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, -- cgit v1.2.3-70-g09d2