mattricesound commited on
Commit
6af03de
·
1 Parent(s): 652f240

Update ckpt paths in chain_inference

Browse files
Files changed (2) hide show
  1. cfg/config.yaml +3 -10
  2. cfg/exp/chain_inference.yaml +5 -5
cfg/config.yaml CHANGED
@@ -51,7 +51,7 @@ datamodule:
51
  _target_: remfx.datasets.EffectDatamodule
52
  train_dataset:
53
  _target_: remfx.datasets.EffectDataset
54
- total_chunks: 80
55
  sample_rate: ${sample_rate}
56
  root: ${oc.env:DATASET_ROOT}
57
  chunk_size: ${chunk_size}
@@ -67,7 +67,7 @@ datamodule:
67
  render_root: ${render_root}
68
  val_dataset:
69
  _target_: remfx.datasets.EffectDataset
70
- total_chunks: 10
71
  sample_rate: ${sample_rate}
72
  root: ${oc.env:DATASET_ROOT}
73
  chunk_size: ${chunk_size}
@@ -83,7 +83,7 @@ datamodule:
83
  render_root: ${render_root}
84
  test_dataset:
85
  _target_: remfx.datasets.EffectDataset
86
- total_chunks: 10
87
  sample_rate: ${sample_rate}
88
  root: ${oc.env:DATASET_ROOT}
89
  chunk_size: ${chunk_size}
@@ -124,10 +124,3 @@ trainer:
124
  devices: 1
125
  gradient_clip_val: 10.0
126
  max_steps: 50000
127
- ckpts:
128
- RandomPedalboardChorus: "/Users/matthewrice/Developer/remfx/ckpts/demucs_5-5/last.ckpt"
129
- RandomPedalboardDelay: "/Users/matthewrice/Developer/remfx/ckpts/demucs_5-5/last.ckpt"
130
- RandomPedalboardDistortion: "/Users/matthewrice/Developer/remfx/ckpts/demucs_5-5/last.ckpt"
131
- RandomPedalboardCompressor: "/Users/matthewrice/Developer/remfx/ckpts/demucs_5-5/last.ckpt"
132
- RandomPedalboardReverb: "/Users/matthewrice/Developer/remfx/ckpts/demucs_5-5/last.ckpt"
133
- num_bins: 1025
 
51
  _target_: remfx.datasets.EffectDatamodule
52
  train_dataset:
53
  _target_: remfx.datasets.EffectDataset
54
+ total_chunks: 8000
55
  sample_rate: ${sample_rate}
56
  root: ${oc.env:DATASET_ROOT}
57
  chunk_size: ${chunk_size}
 
67
  render_root: ${render_root}
68
  val_dataset:
69
  _target_: remfx.datasets.EffectDataset
70
+ total_chunks: 1000
71
  sample_rate: ${sample_rate}
72
  root: ${oc.env:DATASET_ROOT}
73
  chunk_size: ${chunk_size}
 
83
  render_root: ${render_root}
84
  test_dataset:
85
  _target_: remfx.datasets.EffectDataset
86
+ total_chunks: 1000
87
  sample_rate: ${sample_rate}
88
  root: ${oc.env:DATASET_ROOT}
89
  chunk_size: ${chunk_size}
 
124
  devices: 1
125
  gradient_clip_val: 10.0
126
  max_steps: 50000
 
 
 
 
 
 
 
cfg/exp/chain_inference.yaml CHANGED
@@ -27,9 +27,9 @@ datamodule:
27
  batch_size: 16
28
  num_workers: 8
29
  ckpts:
30
- RandomPedalboardChorus: "/Users/matthewrice/Developer/remfx/ckpts/demucs_5-5/last.ckpt"
31
- RandomPedalboardDelay: "/Users/matthewrice/Developer/remfx/ckpts/demucs_5-5/last.ckpt"
32
- RandomPedalboardDistortion: "/Users/matthewrice/Developer/remfx/ckpts/demucs_5-5/last.ckpt"
33
- RandomPedalboardCompressor: "/Users/matthewrice/Developer/remfx/ckpts/demucs_5-5/last.ckpt"
34
- RandomPedalboardReverb: "/Users/matthewrice/Developer/remfx/ckpts/demucs_5-5/last.ckpt"
35
  num_bins: 1025
 
27
  batch_size: 16
28
  num_workers: 8
29
  ckpts:
30
+ RandomPedalboardChorus: "ckpts/chorus.ckpt"
31
+ RandomPedalboardDelay: "ckpts/delay.ckpt"
32
+ RandomPedalboardDistortion: "ckpts/distortion.ckpt"
33
+ RandomPedalboardCompressor: "ckpts/compressor.ckpt"
34
+ RandomPedalboardReverb: "ckpts/reverb.ckpt"
35
  num_bins: 1025