diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-05-04 23:11:44 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-05-04 23:11:44 +0200 |
| commit | 53450493e0a13d835fd1d2457c49a9d60bee0e18 (patch) | |
| tree | ed30fcdc9dca78ef403622842493273dabac3110 /text_recognizer/models/vqvae.py | |
| parent | aa8b4d6089763e8023a1baccb14dd53b6eceea39 (diff) | |
Nyströmer implemented but not tested
Diffstat (limited to 'text_recognizer/models/vqvae.py')
| -rw-r--r-- | text_recognizer/models/vqvae.py | 7 |
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 |