jax.profiler.stop_trace

jax.profiler.stop_trace()

停止当前正在运行的性能分析跟踪。

跟踪将保存到传递给相应 start_trace() 调用的 log_dir。如果尚未开始跟踪，则引发 RuntimeError。