qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2026-07-12 06:39:31 +02:00
Code Issues Projects Releases Wiki Activity

training convert: only add characters_org.txt if it exists

Browse source
This commit is contained in:
Robert Sachunsky 2026-06-26 02:11:59 +02:00
parent 42a3751e63
commit ef47f0ef09
1 changed files with 6 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

3
src/eynollah/training/convert.py
View file

@ -68,8 +68,9 @@ def convert_cli(rebuild, format_, in_, out):
ex.add_config(str(config_path)) ex.add_config(str(config_path))
# some models deviate between training and inference # some models deviate between training and inference
ex.add_config(inference=True) ex.add_config(inference=True)
# make sure the local vocab file gets re-used # OCR models: make sure the local vocab file gets re-used, if available
characters_txt_file = model_path / "characters_org.txt" characters_txt_file = model_path / "characters_org.txt"
if characters_txt_file.exists():
with open(characters_txt_file, "r") as voc_file: with open(characters_txt_file, "r") as voc_file:
voc = json.load(voc_file) voc = json.load(voc_file)
ex.add_config(characters_txt_file=characters_txt_file) ex.add_config(characters_txt_file=characters_txt_file)