Skip to content

Commit 41e1262

Browse files
Curt TiggesCurt Tigges
authored andcommitted
reduced LRU cache size
1 parent f6ef708 commit 41e1262

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

clt/training/data/local_activation_store.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -146,7 +146,7 @@ def _load_norm_stats(self) -> Optional[Dict[str, Any]]:
146146
logger.error(f"Error reading norm_stats file {path}: {e}")
147147
return None
148148

149-
@lru_cache(maxsize=256)
149+
@lru_cache(maxsize=64)
150150
def _load_chunk(self, chunk_path: str, layer_key: str, data_type: str):
151151
"""Loads entire HDF5 chunk from disk and caches"""
152152

0 commit comments

Comments
 (0)