Fix a shape-related bug that came from incorrect usage of unsqueeze

asteroid-team · Oct 12, 2020 · e94bd84 · e94bd84
1 parent ce79409
commit e94bd84
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/demo.py b/scripts/demo.py
@@ -68,7 +68,7 @@ def __exit__(self, type, value, traceback):
     samples, _ = librosa.load(
         os.path.join(TEST_FIXTURES_DIR, "acoustic_guitar_0.wav"), sr=SAMPLE_RATE
     )
-    samples = torch.from_numpy(samples).unsqueeze(1).unsqueeze(1)
+    samples = torch.from_numpy(samples).unsqueeze(0).unsqueeze(0)
 
     transforms = [
         {"instance": Gain(p=1.0), "num_runs": 5},