File size: 229 Bytes
30e9297
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
#!/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}