This test is 'to much' for the resources allocated by GH

Signed-off-by: João Lucas de Sousa Almeida <[email protected]>
IBM · Sep 26, 2023 · 7e67ad1 · 7e67ad1
1 parent 151233f
commit 7e67ad1
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/network/test_template_gen.py b/tests/network/test_template_gen.py
@@ -427,12 +427,12 @@ def test_autoencoder_multiscaleautoencoder(self) -> None:
 
         for device in ["cpu", "gpu", None]:
 
-            input_data = np.random.rand(100, 1, 128, 128)
+            input_data = np.random.rand(100, 1, 32, 32)
 
             for kind_of_ae in ["variational", "vanilla"]:
 
                 autoencoder = MultiScaleAutoencoder(
-                    input_dim=(None, 1, 128, 128),
+                    input_dim=(None, 1, 32, 32),
                     latent_dim=8,
                     kernel_sizes_list=[3,5,7,9],
                     activation="tanh",
@@ -453,7 +453,7 @@ def test_autoencoder_multiscaleautoencoder(self) -> None:
                 estimated_data = autoencoder.reconstruction_forward(input_data=input_data)
                 estimated_data = autoencoder.eval(input_data=input_data)
                 autoencoder.summary()
-
+                del autoencoder
                 assert estimated_data.shape == input_data.shape