From 53450493e0a13d835fd1d2457c49a9d60bee0e18 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Tue, 4 May 2021 23:11:44 +0200
Subject: Nyströmer implemented but not tested
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 text_recognizer/models/vqvae.py | 7 +------
 1 file changed, 1 insertion(+), 6 deletions(-)

(limited to 'text_recognizer/models')

diff --git a/text_recognizer/models/vqvae.py b/text_recognizer/models/vqvae.py
index 078235e..9857420 100644
--- a/text_recognizer/models/vqvae.py
+++ b/text_recognizer/models/vqvae.py
@@ -35,12 +35,7 @@ class LitVQVAEModel(LitBaseModel):
         """Logs prediction on image with wandb."""
         try:
             self.logger.experiment.log(
-                {
-                    title: [
-                        wandb.Image(data[0]),
-                        wandb.Image(reconstructions[0]),
-                    ]
-                }
+                {title: [wandb.Image(data[0]), wandb.Image(reconstructions[0]),]}
             )
         except AttributeError:
             pass
-- 
cgit v1.2.3-70-g09d2