MidiRepGen/dvc.yaml at main · T04STER/MidiRepGen · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
stages:
  download_dataset:
    cmd: python scripts/download_dataset.py
    deps:
      - scripts/download_dataset.py
    params:
      - dataset
    outs:
      - data/midi_dataset.zip
      - data/midi_dataset/
  midi_to_dataset:
    matrix:
      mode: [ 'note_events']
    cmd: python scripts/midi_to_dataset.py --mode ${item.mode}
    deps:
      - scripts/midi_to_dataset.py
      - data/midi_dataset/
    params:
      - dataset
      - preprocess.sample_rate
      - preprocess.note_count
    outs:
      - data/preprocessed_${item.mode}.pkl
      - data/preprocessed_${item.mode}_test.pkl

  train_note_event_vae:
    cmd: python scripts/train_note_event_vae.py
    params:
      - train_note_event_vae.run_name
    deps:
      - scripts/train_note_event_vae.py
      - data/preprocessed_note_events.pkl
    outs:
      - models/${train_note_event_vae.run_name}.pt