From e77508020ad4b33677becd1182d532cf518d9254 Mon Sep 17 00:00:00 2001
From: Chin-Yun Yu <cy006@bluck.eecs.qmul.ac.uk>
Date: Tue, 2 May 2023 10:24:35 +0100
Subject: [PATCH] re-test wide pitch and speed augmentation 30k steps

---
 .../hdemucs-64-sdr/checkpoints/last.ckpt      |  4 +-
 .../lightning_logs/hdemucs-64-sdr/config.yaml | 40 ++++++++++++++++++-
 2 files changed, 41 insertions(+), 3 deletions(-)

diff --git a/my_submission/lightning_logs/hdemucs-64-sdr/checkpoints/last.ckpt b/my_submission/lightning_logs/hdemucs-64-sdr/checkpoints/last.ckpt
index a2de535..71d5d11 100644
--- a/my_submission/lightning_logs/hdemucs-64-sdr/checkpoints/last.ckpt
+++ b/my_submission/lightning_logs/hdemucs-64-sdr/checkpoints/last.ckpt
@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b37989dc9b3c321b1ea016b91225acf8e91efc2a5c88c1522b33fac21954a3b4
-size 594510557
+oid sha256:87495f5b34882e5a7e03472ca7d2650a7b65bd1e52884fbdeb488d86cefbe4dd
+size 594681571
diff --git a/my_submission/lightning_logs/hdemucs-64-sdr/config.yaml b/my_submission/lightning_logs/hdemucs-64-sdr/config.yaml
index 6cb3b5b..4b6b4b1 100644
--- a/my_submission/lightning_logs/hdemucs-64-sdr/config.yaml
+++ b/my_submission/lightning_logs/hdemucs-64-sdr/config.yaml
@@ -118,7 +118,45 @@ model:
         dconv_init: 0.0001
     criterion:
       class_path: aimless.loss.time.NegativeSDR
-    transforms: null
+    transforms:
+    - class_path: aimless.augment.SpeedPerturb
+      init_args:
+        orig_freq: 44100
+        speeds:
+        - 50
+        - 60
+        - 70
+        - 80
+        - 90
+        - 100
+        - 110
+        - 120
+        - 130
+        - 140
+        - 150
+        p: 0.3
+    - class_path: aimless.augment.RandomPitch
+      init_args:
+        semitones:
+        - -7
+        - -6
+        - -5
+        - -4
+        - -3
+        - -2
+        - -1
+        - 1
+        - 0
+        - 1
+        - 2
+        - 3
+        - 4
+        - 5
+        - 6
+        - 7
+        n_fft: 2048
+        hop_length: 512
+        p: 0.3
     use_sdx_targets: false
     target_track: sdx
     targets:
-- 
GitLab