From 61b6934f914b1615483a255419ac8471853901e9 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
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/models/base.py')

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