Update ein_fft.py

kyegomez · Mar 26, 2024 · 4bb33eb · 4bb33eb
1 parent c638864
commit 4bb33eb
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/simba_torch/ein_fft.py b/simba_torch/ein_fft.py
@@ -195,11 +195,11 @@ def forward(self, x: Tensor) -> Tensor:
 
 
 # Random input tensor
-x = torch.randn(1, 3, 64)
+# x = torch.randn(1, 3, 64)
 
-# Instantiate EinFFT module
-einfft = EinFFTText(3, 64)
+# # Instantiate EinFFT module
+# einfft = EinFFTText(3, 64)
 
-# Apply EinFFT to get an output
-output = einfft(x)
-print(output.shape)
+# # Apply EinFFT to get an output
+# output = einfft(x)
+# print(output.shape)