| from TranslatorTrainer import * | |
| vocab_filepaths = ( | |
| 'data/lsr2_phoneme_vocab.pth', | |
| 'data/lsr2_text_char_vocab.pth' | |
| ) | |
| trainer = TranslatorTrainer( | |
| word_tokenize=False, | |
| input_char_map=options.char_map, | |
| output_char_map=options.text_char_map, | |
| dataset_type=Datasets.LRS2 | |
| ) | |
| # trainer.train() | |
| trainer.save_vocabs(*vocab_filepaths) |