thai_indextts2 / train.bat
williampike's picture
Upload folder using huggingface_hub
4d3248c verified
Raw
History Blame Contribute Delete
652 Bytes
uv run python trainers/train_gpt_v2.py ^
--train-manifest omniman2_data/train_ipa.jsonl ^
--val-manifest omniman2_data/val_ipa.jsonl ^
--tokenizer checkpoints/thaiipabpe.model ^
--config checkpoints/config.yaml ^
--base-checkpoint checkpoints/gpt.pth ^
--output-dir checkpoints/thai_gpt_v5 ^
--batch-size 16 ^
--grad-accumulation 1 ^
--epochs 2 ^
--learning-rate 1e-5 ^
--weight-decay 0.01 ^
--warmup-steps 1000 ^
--log-interval 1 ^
--val-interval 1000 ^
--grad-clip 1.0 ^
--text-loss-weight 0.2 ^
--mel-loss-weight 0.8 ^
--amp ^
--use-duration-control