diff options
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: |