Spaces:

descript
/

vampnet

Sleeping

Hugo Flores commited on Mar 22, 2023

Commit

d6a029b

1 Parent(s): 920f55c

fix transforms

Files changed (1) hide show

scripts/exp/train.py CHANGED Viewed

@@ -59,7 +59,8 @@ IGNORE_INDEX = -100
 @argbind.bind("train", "val", without_prefix=True)
 def build_transform():
     transform = transforms.Compose(
-        tfm.VolumeNorm(("uniform", -24)),
     )
     return transform

 @argbind.bind("train", "val", without_prefix=True)
 def build_transform():
     transform = transforms.Compose(
+        tfm.VolumeNorm(("const", -24)),
+        tfm.RescaleAudio(),
     )
     return transform