From 793f1ee1a653dd6a4eb47dab357aa5c0e2a9eb72 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Sun, 10 Oct 2021 18:07:40 +0200 Subject: Refactor of vq models --- text_recognizer/models/vqgan.py | 1 - 1 file changed, 1 deletion(-) (limited to 'text_recognizer/models/vqgan.py') 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: -- cgit v1.2.3-70-g09d2