Skip to content
Snippets Groups Projects
Commit 45b2a05c authored by Maria Guaranda-Cabezas's avatar Maria Guaranda-Cabezas
Browse files

adds changes for another experiment

parent c7040c8c
No related branches found
No related tags found
1 merge request!3Inference
......@@ -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)
......
......@@ -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)
......
......@@ -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
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment