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

Changed test to use pool trainer

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

4
Project_Model/Tests/bpe_trainer_test.py
View File

@ -4,13 +4,13 @@ import Project_Model.Libs.BPE as BPE
import re import re
CACHE_DIR_PATH = Path("Project_Model/Tests/trainer_files/cache") CACHE_DIR_PATH = Path("Project_Model/Tests/trainer_files/cache/pool-cache.json")
class TestTrainBPE: class TestTrainBPE:
def test_bpe_train_encoding_simple(self): def test_bpe_train_encoding_simple(self):
TRAINER = BPE.NanoSocraTrainer( TRAINER = BPE.NanoSocraTrainerPool(
int(32E3), int(32E3),
["<SOT>", "<EOT>"], ["<SOT>", "<EOT>"],
40 40