Voltline's picture
Release VimeML V2.1 step40000 FP32 and Core ML INT8 (GPL-2.0)
29f25be verified
Raw History Blame Contribute Delete
387 Bytes
"""Child process capturing SentencePiece's native C++ training diagnostics."""
import json
import sys
from pathlib import Path
def main():
import sentencepiece as spm
request = json.loads(Path(sys.argv[1]).read_text(encoding='utf-8'))
spm.set_random_generator_seed(request['seed'])
spm.SentencePieceTrainer.train(**request['options'])
if __name__=='__main__': main()