feat: invalidate cache if requirements change

meakbiyik · Sep 1, 2023 · 18eb48e · 18eb48e
1 parent 8264279
commit 18eb48e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/torchcache/torchcache.py b/torchcache/torchcache.py
@@ -343,6 +343,8 @@ def wrap_module(
                 logger.error(f"Could not retrieve the module source: {e}")
                 # If the module source cannot be retrieved, we use the module name
                 hash_string = module.__class__.__name__ + repr(args) + repr(kwargs)
+            # Also add the crucial parameters of torchcache
+            hash_string += repr(self.subsample_count) + repr(self.zstd_compression)
             logger.debug(f"Module hash string: {hash_string}")
             self.module_hash = hashlib.blake2b(
                 hash_string.encode(),