Merge pull request #1 from lightly-ai/low_memory_tests_fix_IS

Low memory causes tests fail fix

Author: IgorSusmelj

Diff for: .gitignore

@@ -1,3 +1,4 @@
 **__pycache__**
 **build
-**settings.json
+**settings.json
+lightning_logs/

Diff for: tests/models/test_ModelsSimCLR.py

@@ -15,7 +15,7 @@ def setUp(self):
             'resnet-101',
             'resnet-152'
         ]
-        self.batch_size = 4
+        self.batch_size = 2
         self.input_tensor = torch.rand((self.batch_size, 3, 32, 32))
 
     def test_create_variations_cpu(self):
@@ -59,8 +59,8 @@ def test_feature_dim_configurable(self):
     def test_variations_input_dimension(self):
         device = 'cuda' if torch.cuda.is_available() else 'cpu'
         for model_name in self.resnet_variants:
-            for input_width in [16, 32, 64, 256]:
-                for input_height in [16, 32, 64, 256]:
+            for input_width in [16, 32, 64, 128]:
+                for input_height in [16, 32, 64, 128]:
                     model = ResNetSimCLR(name=model_name).to(device)
 
                     input_tensor = torch.rand((self.batch_size,