#!/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" \
  "$@"