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}