jax.errors.UnexpectedTracerError 仅在使用 jax.debug.breakpoint（）时出错-解网

问：

我的 jax 代码运行良好，但是当我尝试使用 jax.debug.breakpoint 插入断点时，出现错误：jax.errors.UnexpectedTracerError。

我希望在不设置断点的情况下也会出现此错误。

这是有意的行为还是发生了一些奇怪的事情？使用jax_checking_leaks时，报告的示踪剂似乎都没有真正泄漏。

python 调试断点 jax google-jax

答：

1赞 jakevdp 9/8/2023 #1

目前存在一个错误，在某些情况下可能导致虚假示踪剂泄漏：请参阅 https://github.com/google/jax/issues/16732。jax.debug.breakpoint

不幸的是，目前没有任何简单的解决方法，但希望这个问题能很快得到解决。

jax.errors.UnexpectedTracerError 仅在使用 jax.debug.breakpoint（）时出错