PoliBa-DeepLearning/NanoSocrates
2
0
Fork 0
Code Issues 3 Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Changed training phase to take directly data instead of its encode

Browse Source
This commit is contained in:
Christian Risi 2025-10-02 09:56:44 +02:00
parent 2194cc7b4f
commit 2e595a3a23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
Project_Model/Libs/BPE/Classes/NanoSocraTrainerPool.py
View File

@ -46,7 +46,7 @@ def split_fit(object: tuple[NanoSocratesBPE, list[list[int]]]):
continue
# We are sure of its type
NEW_DATA.append(output) # type: ignore
NEW_DATA.append(piece) # type: ignore
return (bpe, NEW_DATA, memory)
@ -56,14 +56,14 @@ def split_encode(object: tuple[NanoSocratesBPE, list[list[int]]]):
NEW_DATA: list[list[int]] = []
for piece in data:
for index, piece in zip(range(0, len(data)), data):
output = bpe.encode_intermediate(piece)
if len(output) < 2:
continue
# We are sure of its type
NEW_DATA.append(output) # type: ignore
NEW_DATA.append(data[index]) # type: ignore
return NEW_DATA