Skip to content

Latest commit

 

History

History
26 lines (21 loc) · 551 Bytes

README.md

File metadata and controls

26 lines (21 loc) · 551 Bytes

aivivn-vn-tones

Generate data

Generate data for default problem translate_vndt

./gen_data.sh

Generate data for custom problem A

./gen_data.sh A

Train model

On problem translate_vndt, to train model transformer with hparams transformer_base on GPUs 0,1

./train.sh 0,1 transformer_base transformer translate_vndt

Predict

Similar to train.sh

./predict.sh 0,1 transformer_base transformer translate_vndt

The output is stored in sub-translate_vndt-transformer-transformer_base.csv