Update configs/Kiritan.py

configs/Kiritan.py

@@ -1,15 +1,8 @@
-from pytorch_lightning.callbacks import LearningRateMonitor, ModelCheckpoint
-
 _base_ = [
     "./_base_/archs/hifi_svc.py",
-    "./_base_/trainers/base.py",
-    "./_base_/schedulers/exponential.py",
-    "./_base_/datasets/hifi_svc.py",
 ]
 
-speaker_mapping = {
-    "kiritan": 0,
-}
+speaker_mapping = {'kiritan': 0,}
 
 model = dict(
     type="HiFiSVC",
@@ -33,24 +26,14 @@ preprocessing = dict(
     ),
     augmentations=[
         dict(
-            type="FixedPitchShifting",
-            key_shifts=[-5.0, 5.0],
-            probability=0.75,
-        ),
-    ],
-)
-
-trainer = dict(
-    # Disable gradient clipping, which is not supported by custom optimization
-    gradient_clip_val=None,
-    val_check_interval=1000,
-    check_val_every_n_epoch=None,
-    callbacks=[
-        ModelCheckpoint(
-            filename="{epoch}-{step}-{valid_loss:.2f}",
-            every_n_train_steps=1000,
-            save_top_k=-1,
+            type="RandomPitchShifting",
+            key_shifts=[-5., 5.],
+            probability=1.5,
         ),
-        LearningRateMonitor(logging_interval="step"),
+        dict(
+            type="RandomTimeStretching",
+            factors=[0.8, 1.2],
+            probability=0.75,
+        )
     ],
 )