jax.profiler.stop_trace

jax.profiler.stop_trace()

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

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