code: def __init__(self, module, size=1):
      super().__init__()
      if size != 1:
          raise ValueError("We do not support LRUCache bigger than 1.")
      self.module = module
      self._memory_input = None
      self._memory_output = None
  
file_path: baal/baal/bayesian/caching_utils.py
imports: from typing import Optional
  import torch
  from torch import nn, Tensor
  from baal.bayesian.common import BayesianModule, _patching_wrapper
label_desc: Use image node_id 1 to create a new LRUCacheModule object from inherited base classes: nn with example: obj = LRUCacheModule(module, size)