jax.profiler.stop_trace#
- jax.profiler.stop_trace()[源代码][源代码]#
停止当前正在运行的分析器跟踪。
跟踪将被保存到传递给
start_trace()调用的log_dir中。如果尚未开始跟踪，则会引发 RuntimeError。
停止当前正在运行的分析器跟踪。
跟踪将被保存到传递给 start_trace() 调用的 log_dir 中。如果尚未开始跟踪，则会引发 RuntimeError。