Fixed a masking problem

2025-10-14 10:34:14 +02:00 · 2025-10-14 10:34:14 +02:00 · 4968d79403
commit 4968d79403
parent 80fd7fd600
1 changed files with 2 additions and 2 deletions
--- a/Project_Model/Libs/Batch/Classes/Batcher.py
+++ b/Project_Model/Libs/Batch/Classes/Batcher.py
@ -157,7 +157,7 @@ class Batcher:
        X = []
        Y = []
        for rdf in batch["RDFs"]:
-            x, y = self._masker.mask_sequence(rdf)
+            x, y = self._masker.mask_sequence(rdf[:self.__max_length])
            X.append(x)
            Y.append(y)
        return self.__normalization(X, Y)