Reduce batch and image size for model tests

The old test setup failed on my 10GB ram machine due to
out of memory issues.

Author: IgorSusmelj

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,