jax.checking_leaks

jax.checking_leaks = functools.partial(<jax._src.config.State object>, True)

jax_check_tracer_leaks 設定選項的上下文管理器。

在追蹤完成後立即開啟檢查洩漏的追蹤器。啟用洩漏檢查可能會對效能產生影響：某些快取會停用，並且可能會增加其他額外負荷。此外，請注意某些 Python 除錯器可能會導致誤報，因此建議在啟用洩漏檢查時停用任何除錯器。