add logtext

tuanio · Oct 25, 2022 · b405eef · b405eef
1 parent 28304ea
commit b405eef
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/image2latex/model.py b/image2latex/model.py
@@ -203,7 +203,7 @@ def test_step(self, batch, batch_idx):
             )
         )
 
-        if self.log_text and batch_idx % self.log_step == 0:
+        if True and batch_idx % self.log_step == 0:
             for truth, pred in zip(truths, predicts):
                 print("=" * 20)
                 print(f"Truth: [{' '.join(truth)}] | Predict: [{' '.join(pred)}]")