#!/bin/bash
# Train the music generation model
set -e
cd "$(dirname "$0")/.."

python -m src.s00_main train \
    --epochs ${EPOCHS:-50} \
    --batch-size ${BATCH_SIZE:-8} \
    --lr ${LR:-3e-4} \
    --seq-len ${SEQ_LEN:-1024}