From facce1b73420b645de7bb07f01fffc9de34febb8 Mon Sep 17 00:00:00 2001 From: Chaojian LI Date: Wed, 5 May 2021 04:48:39 -0500 Subject: [PATCH] on_trace_ready is in torch.profiler.profile on_trace_ready should be the the arg of [torch.profiler.profile](https://pytorch.org/docs/stable/profiler.html#torch.profiler.profile) instead of [torch.profiler.schedule](https://pytorch.org/docs/stable/profiler.html#torch.profiler.schedule). --- tb_plugin/README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tb_plugin/README.md b/tb_plugin/README.md index 051e910c9..d63fc6f9c 100644 --- a/tb_plugin/README.md +++ b/tb_plugin/README.md @@ -40,7 +40,7 @@ and give optimization recommendations. or learn from [PyTorch Profiler](https://pytorch.org/tutorials/recipes/recipes/profiler_recipe.html). Note: The recommended way to produce profiling data is assigning "torch.profiler.tensorboard_trace_handler" - to "on_trace_ready" on creation of "torch.profiler.schedule". + to "on_trace_ready" on creation of "torch.profiler.profile". * Start TensorBoard