Add example python script with initialize_globals_from_args()

This commit is contained in:
Josh Meyer 2021-07-21 15:13:39 -04:00
parent ae9280ef1a
commit ec37b3324a

27
bin/run-ldc93s1.py Executable file
View File

@ -0,0 +1,27 @@
#!/usr/bin/env python
import os
from import_ldc93s1 import _download_and_preprocess_data as download_ldc
from coqui_stt_training.util.config import Config, initialize_globals_from_args
from coqui_stt_training.train import train, test, early_training_checks
from coqui_stt_ctcdecoder import Alphabet
import tensorflow.compat.v1 as tfv1
# only one GPU for only one training sample
os.environ['CUDA_VISIBLE_DEVICES']='0'
download_ldc("data/ldc93s1")
initialize_globals_from_args(
alphabet_config_path="data/alphabet.txt",
train_files=["data/ldc93s1/ldc93s1.csv"],
dev_files=["data/ldc93s1/ldc93s1.csv"],
test_files=["data/ldc93s1/ldc93s1.csv"],
n_hidden=100,
epochs=200
)
early_training_checks()
train()
tfv1.reset_default_graph()
test()