32 lines
1.2 KiB
Bash
32 lines
1.2 KiB
Bash
#!/usr/bin/env bash
|
|
# Run inside Ubuntu after the isolated Piper environment and dataset are prepared.
|
|
set -euo pipefail
|
|
cd "$HOME/teddy-training/piper"
|
|
source "$HOME/teddy-training/venv/bin/activate"
|
|
export OMP_NUM_THREADS=4
|
|
python /mnt/d/Projects/escape-gallery/room_glitch/labs/glitch_odin/train_teddy_entry.py fit \
|
|
--seed_everything 42 \
|
|
--data.voice_name teddy_experiment_v1 \
|
|
--data.csv_path "$HOME/teddy-training/data/metadata.csv" \
|
|
--data.audio_dir "$HOME/teddy-training/data/audio" \
|
|
--data.cache_dir "$HOME/teddy-training/cache" \
|
|
--data.config_path "$HOME/teddy-training/teddy-config.json" \
|
|
--data.espeak_voice en-us \
|
|
--data.batch_size 4 \
|
|
--data.num_workers 0 \
|
|
--data.validation_split 0.1 \
|
|
--data.num_test_examples 2 \
|
|
--model.sample_rate 22050 \
|
|
--model.warmstart_ckpt "$HOME/teddy-training/checkpoints/ljspeech-medium.ckpt" \
|
|
--model.learning_rate 0.0001 \
|
|
--model.learning_rate_d 0.00005 \
|
|
--model.mos_metric none \
|
|
--trainer.accelerator gpu \
|
|
--trainer.devices 1 \
|
|
--trainer.max_epochs 60 \
|
|
--trainer.max_steps 1000 \
|
|
--trainer.enable_progress_bar false \
|
|
--trainer.log_every_n_steps 5 \
|
|
--trainer.default_root_dir "$HOME/teddy-training/runs/teddy-v1" \
|
|
"$@"
|