H1K0
dd4f21f17f
- scripts/run_pretrain.py: single-command pre-training runner with timing estimate, loss-curve plot (matplotlib), and per-epoch report. Sets max_seq_len=256 (McGill sequences max out at 195 tokens, ~4x faster attention than the 512 default). - src/train.py: normalise --output so pretrained.pt and pretrained both produce pretrained.pt + pretrained.log.csv (not pretrained.pt.log.csv). Serialize Path fields as strings in checkpoint to satisfy weights_only. - requirements.txt: drop unused pandas/music21, add mido (pretty_midi dep). Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
17 lines
204 B
Plaintext
17 lines
204 B
Plaintext
# Python >= 3.11 required
|
|
# Tested on Python 3.12.10
|
|
|
|
# Core ML
|
|
torch==2.12.0
|
|
numpy==2.4.6
|
|
|
|
# Music processing
|
|
pretty_midi==0.2.11
|
|
mido==1.3.3
|
|
|
|
# Visualization
|
|
matplotlib==3.10.9
|
|
|
|
# Testing
|
|
pytest==9.0.3
|