test

Author: jacobgil

setup.py

@@ -8,7 +8,7 @@
 
 setuptools.setup(
     name='grad-cam',
-    version='1.4.6',
+    version='1.4.8',
     author='Jacob Gildenblat',
     author_email='[email protected]',
     description='Many Class Activation Map methods implemented in Pytorch for classification, segmentation, object detection and more',

tests/test_context_release.py

@@ -66,4 +66,4 @@ def test_memory_usage_in_loop(numpy_image, batch_size, width, height,
 
             if i == 0:
                 initial_memory = psutil.virtual_memory()[2]
-        assert(psutil.virtual_memory()[2] <= initial_memory * 1.05)
+        assert(psutil.virtual_memory()[2] <= initial_memory * 1.1)