Spaces:
Runtime error
Runtime error
File size: 1,789 Bytes
6da1b0d 634cc2a 6da1b0d a559a3b 6da1b0d fb9ce8b 6da1b0d ace4057 6da1b0d ace4057 fb9ce8b 133e1dc |
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 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 |
# @package _global_
defaults:
- override /model: demucs
- override /effects: all
seed: 12345
sample_rate: 48000
chunk_size: 262144 # 5.5s
logs_dir: "./logs"
accelerator: "gpu"
log_audio: True
# Effects
num_kept_effects: [0,0] # [min, max]
num_removed_effects: [0,5] # [min, max]
shuffle_kept_effects: True
shuffle_removed_effects: True
num_classes: 5
effects_to_keep:
effects_to_remove:
- distortion
- compressor
- reverb
- chorus
- delay
datamodule:
train_batch_size: 1
test_batch_size: 1
num_workers: 8
train_dataset: None
val_dataset: None
test_dataset:
_target_: remfx.datasets.InferenceDataset
root: ${oc.env:DATASET_ROOT}
sample_rate: ${sample_rate}
dcunet:
_target_: remfx.models.RemFX
lr: 1e-4
lr_beta1: 0.95
lr_beta2: 0.999
lr_eps: 1e-6
lr_weight_decay: 1e-3
sample_rate: ${sample_rate}
network:
_target_: remfx.models.DCUNetModel
architecture: "Large-DCUNet-20"
stft_kernel_size: 512
fix_length_mode: "pad"
sample_rate: ${sample_rate}
num_bins: 1025
ckpts:
RandomPedalboardDistortion:
model: ${model}
ckpt_path: "ckpts/demucs_distortion_aug.ckpt"
RandomPedalboardCompressor:
model: ${model}
ckpt_path: "ckpts/demucs_compressor_aug.ckpt"
RandomPedalboardReverb:
model: ${dcunet}
ckpt_path: "ckpts/dcunet_reverb_aug.ckpt"
RandomPedalboardChorus:
model: ${dcunet}
ckpt_path: "ckpts/dcunet_chorus_aug.ckpt"
RandomPedalboardDelay:
model: ${dcunet}
ckpt_path: "ckpts/dcunet_delay_aug.ckpt"
inference_effects_ordering:
- "RandomPedalboardDistortion"
- "RandomPedalboardCompressor"
- "RandomPedalboardReverb"
- "RandomPedalboardChorus"
- "RandomPedalboardDelay"
num_bins: 1025
inference_effects_shuffle: False
inference_use_all_effect_models: False |