diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-10-10 18:07:40 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-10-10 18:07:40 +0200 |
| commit | 793f1ee1a653dd6a4eb47dab357aa5c0e2a9eb72 (patch) | |
| tree | 6b64398a7ab55a4e3d1acccfed46c25d9bd80ea7 /text_recognizer/models/vqgan.py | |
| parent | ef5e7e6a62a6d7223fe5a5137e2bfacba9ec12b6 (diff) | |
Refactor of vq models
Diffstat (limited to 'text_recognizer/models/vqgan.py')
| -rw-r--r-- | text_recognizer/models/vqgan.py | 1 |
1 files changed, 0 insertions, 1 deletions
diff --git a/text_recognizer/models/vqgan.py b/text_recognizer/models/vqgan.py index 30b46a7..6e4d3fb 100644 --- a/text_recognizer/models/vqgan.py +++ b/text_recognizer/models/vqgan.py @@ -24,7 +24,6 @@ class VQGANLitModel(BaseLitModel): ) -> Tensor: """Training step.""" data, _ = batch - reconstructions, commitment_loss = self(data) if optimizer_idx == 0: |