convert size to an integer

mindspore-lab · May 27, 2024 · 960c574 · 960c574
1 parent dac2427
commit 960c574
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mindaudio/models/decoders/greedydecoder.py b/mindaudio/models/decoders/greedydecoder.py
@@ -106,7 +106,7 @@ def process_string(self, sequence, size, remove_repetitions=False):
         """
         string = ""
         offsets = []
-        for i in range(size):
+        for i in range(int(size)):
             char = self.int_to_char[sequence[i].item()]
             if char != self.int_to_char[self.blank_index]:
                 if (