summaryrefslogtreecommitdiff
path: root/text_recognizer/models/vqvae.py
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2021-05-04 23:11:44 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2021-05-04 23:11:44 +0200
commit53450493e0a13d835fd1d2457c49a9d60bee0e18 (patch)
treeed30fcdc9dca78ef403622842493273dabac3110 /text_recognizer/models/vqvae.py
parentaa8b4d6089763e8023a1baccb14dd53b6eceea39 (diff)
Nyströmer implemented but not tested
Diffstat (limited to 'text_recognizer/models/vqvae.py')
-rw-r--r--text_recognizer/models/vqvae.py7
1 files changed, 1 insertions, 6 deletions
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