uploaded drums model

drums.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1c6e6f41b5eca279af8f6500a8d6c193f6968598fb0cb0ba33c317323cc8acbc
+size 521819269

hparams.yaml

@@ -0,0 +1,158 @@
+model:
+  sr: 44100
+  n_fft: 2048
+  bandsplits:
+  - - 1000
+    - 50
+  - - 2000
+    - 100
+  - - 4000
+    - 250
+  - - 8000
+    - 500
+  - - 16000
+    - 1000
+  bottleneck_layer: rnn
+  t_timesteps: 263
+  fc_dim: 128
+  rnn_dim: 256
+  rnn_type: LSTM
+  bidirectional: true
+  num_layers: 10
+  mlp_dim: 512
+  return_mask: false
+  complex_as_channel: true
+  is_mono: false
+train_dataset:
+  file_dir: /home/afteraugustmusician/Music-Demixing-with-Band-Split-RNN/datasets/Drums
+  txt_dir: files/
+  txt_path: null
+  target: drums
+  is_training: true
+  is_mono: false
+  sr: 44100
+  preload_dataset: false
+  silent_prob: 0.1
+  mix_prob: 0.25
+  mix_tgt_too: false
+test_dataset:
+  in_fp: /home/afteraugustmusician/Music-Demixing-with-Band-Split-RNN/datasets/Drums
+  target: drums
+  is_mono: false
+  sr: 44100
+  win_size: 3
+  hop_size: 0.5
+  batch_size: 4
+  window: null
+sad:
+  sr: 44100
+  window_size_in_sec: 6
+  overlap_ratio: 0.5
+  n_chunks_per_segment: 10
+  eps: 1.0e-05
+  gamma: 0.001
+  threshold_max_quantile: 0.15
+  threshold_segment: 0.5
+augmentations:
+  randomcrop:
+    _target_: data.augmentations.RandomCrop
+    p: 1
+    chunk_size_sec: 3
+    sr: 44100
+    window_stft: 2048
+    hop_stft: 512
+  gainscale:
+    _target_: data.augmentations.GainScale
+    p: 0.5
+    min_db: -10.0
+    max_db: 10.0
+featurizer:
+  direct_transform:
+    _target_: torchaudio.transforms.Spectrogram
+    n_fft: 2048
+    win_length: 2048
+    hop_length: 512
+    power: null
+  inverse_transform:
+    _target_: torchaudio.transforms.InverseSpectrogram
+    n_fft: 2048
+    win_length: 2048
+    hop_length: 512
+callbacks:
+  lr_monitor:
+    _target_: pytorch_lightning.callbacks.LearningRateMonitor
+    logging_interval: epoch
+  model_ckpt:
+    _target_: pytorch_lightning.callbacks.ModelCheckpoint
+    monitor: train/loss
+    mode: min
+    save_top_k: 5
+    dirpath: /home/afteraugustmusician/Music-Demixing-with-Band-Split-RNN/src/logs/bandsplitrnn/2023-05-08_15-35/weights
+    filename: epoch{epoch:02d}-train_loss{train/loss:.2f}
+    auto_insert_metric_name: false
+  model_ckpt_usdr:
+    _target_: pytorch_lightning.callbacks.ModelCheckpoint
+    monitor: train/usdr
+    mode: max
+    save_top_k: 5
+    dirpath: /home/afteraugustmusician/Music-Demixing-with-Band-Split-RNN/src/logs/bandsplitrnn/2023-05-08_15-35/weights
+    filename: epoch{epoch:02d}-train_usdr{train/usdr:.2f}
+    auto_insert_metric_name: false
+  ema:
+    _target_: utils.callbacks.EMA
+    decay: 0.9999
+    validate_original_weights: false
+    every_n_steps: 1
+logger:
+  tensorboard:
+    _target_: pytorch_lightning.loggers.TensorBoardLogger
+    save_dir: /home/afteraugustmusician/Music-Demixing-with-Band-Split-RNN/src/logs/bandsplitrnn/2023-05-08_15-35/tb_logs
+    name: ''
+    version: ''
+    log_graph: false
+    default_hp_metric: false
+    prefix: ''
+  wandb:
+    _target_: pytorch_lightning.loggers.WandbLogger
+    project: MDX_BSRNN_23
+    name: drums
+    save_dir: wandb_logs
+    offline: false
+    id: null
+    log_model: false
+    prefix: ''
+    job_type: train
+    group: ''
+    tags: []
+train_loader:
+  batch_size: 8
+  num_workers: 12
+  shuffle: true
+  drop_last: true
+val_loader:
+  batch_size: 2
+  num_workers: 8
+  shuffle: false
+  drop_last: false
+opt:
+  _target_: torch.optim.Adam
+  lr: 0.001
+sch:
+  warmup_step: 10
+  alpha: 1
+  gamma: 0.9899494936611665
+ckpt_path: logs/bandsplitrnn/2023-04-29_14-45/weights/drums-193-usdr-5.29.ckpt
+trainer:
+  fast_dev_run: false
+  min_epochs: 100
+  max_epochs: 500
+  log_every_n_steps: 10
+  accelerator: auto
+  devices: auto
+  gradient_clip_val: 5
+  precision: 32
+  enable_progress_bar: true
+  benchmark: true
+  deterministic: false
+experiment_dirname: bandsplitrnn
+wandb_api_key: d5c4447e39b2b10b95f05f907d57845ded16bc13