jax.profiler.stop_trace#

jax.profiler.stop_trace()[原始碼]#

停止目前正在執行的效能分析追蹤。

追蹤將儲存到傳遞給對應 start_trace() 呼叫的 log_dir。如果尚未啟動追蹤，則會引發 RuntimeError。