Function clear_torch_cache

lm-eval-harness/lm_eval/models/utils.py:191–193 · view source on GitHub ↗

()

Source from the content-addressed store, hash-verified

189
190
191	def clear_torch_cache() -> None:
192	gc.collect()
193	torch.cuda.empty_cache()
194
195
196	def get_dtype(dtype: Union[str, torch.dtype]) -> torch.dtype:

_detect_batch_sizeMethod · 0.90

no outgoing calls

no test coverage detected