Batcher added

2025-10-10 20:10:08 +02:00
parent bed9718f27
commit 96610612fe
3 changed files with 119 additions and 54 deletions
--- a/Project_Model/Libs/Transformer/Classes/SpannedMasker.py
+++ b/Project_Model/Libs/Transformer/Classes/SpannedMasker.py
@@ -25,6 +25,11 @@ class SpannedMasker:
        self.__forbidden_tokens = forbidden_tokens


+    def reseed(self, seed:int):
+        self.__rng = random.Random(seed)
+
+
+
    def mask_sequence(
        self,
        token_sequence: list[int],