From 14db9eb9659240314ac85470cb64143a46593891 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Sol=C3=A8ne=20Tarride?= <starride@teklia.com> Date: Tue, 19 Sep 2023 12:47:17 +0200 Subject: [PATCH] Remove debug code --- dan/ocr/decoder.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/dan/ocr/decoder.py b/dan/ocr/decoder.py index bd1044d7..6f635ee0 100644 --- a/dan/ocr/decoder.py +++ b/dan/ocr/decoder.py @@ -516,8 +516,6 @@ class CTCLanguageDecoder: Add CTC frames between each characters to avoid duplicate removal """ batch_size, _, n_tokens = batch_features.shape - torch.clone(batch_features) - # visualize_debug(batch_features.exp()[0, :batch_frames[0], :].numpy(), "probs.jpg", False) # Create tensor with high probability CTC token high_prob = 0.99 -- GitLab