adds changes for another experiment

src/diffwave/dataset.py

@@ -153,7 +153,6 @@ def from_path(args, params, is_distributed=False):
       # in an exploration notebook
       dataset = ParticleDatasetVx(path = data_dirs[0], 
                                   transform=transforms.Compose([StandardScaler(mean=-0.0003, std=1.7358), 
-                                                                ScaleDiffusionRange(),
                                                                 ToDiffwaveTensor()]))
     else: #with condition
       dataset = ConditionalDataset(data_dirs)

src/diffwave/inference.py

@@ -111,7 +111,6 @@ def main(args):
       # this is a lagrangian trajectory, we have to apply the inverse of 
       # the transformations used when preprocessing
       reverse_transform = Compose([
-        Lambda(lambda t: (t + 1) / 2),
         Lambda(lambda t: t.numpy(force=True).astype(np.float64).transpose()),
       ])
       trajectory = reverse_transform(audio)

src/diffwave/params.py

@@ -35,7 +35,7 @@ class AttrDict(dict):
 params = AttrDict(
     # Training params
     batch_size=64,
-    learning_rate=2e-4,
+    learning_rate=1e-5,
     max_grad_norm=None,
 
     # Data params