File size: 827 Bytes
297c44b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 |
import sys
import speechbrain as sb
from hyperpyyaml import load_hyperpyyaml
from speechbrain.utils.distributed import run_on_main
if __name__ == "__main__":
hparams_file_path, run_opts, overrides = sb.parse_arguments(sys.argv[1:])
with open(hparams_file_path) as hparams_file:
hparams = load_hyperpyyaml(hparams_file, overrides)
sb.create_experiment_directory(
experiment_directory=hparams["output_folder"],
hyperparams_to_save=hparams_file_path,
overrides=overrides,
)
from matbn_prepare import prepare_matbn
run_on_main(
prepare_matbn,
kwargs={
"dataset_folder": hparams["dataset_folder"],
"save_folder": hparams["prepare_folder"],
"keep_unk": hparams["keep_unk"],
},
)
hparams["tokenizer"]()
|