diff --git a/beginner_source/profiler.py b/beginner_source/profiler.py index 0b69009720..6a3894cbbd 100644 --- a/beginner_source/profiler.py +++ b/beginner_source/profiler.py @@ -319,3 +319,4 @@ def forward(self, input, mask): # - `Profiler Usage Recipe `__ # - `Profiling RPC-Based Workloads `__ # - `Profiler API Docs `__ +# - `Hugging Face PyTorch Profiler Blog Post `__ diff --git a/recipes_source/recipes/profiler_recipe.py b/recipes_source/recipes/profiler_recipe.py index a8d1a4dc6b..ef0471e75d 100644 --- a/recipes_source/recipes/profiler_recipe.py +++ b/recipes_source/recipes/profiler_recipe.py @@ -480,4 +480,4 @@ def trace_handler(p): # # - `PyTorch Benchmark `_ # - `Visualizing models, data, and training with TensorBoard `_ tutorial -# +# - `Hugging Face PyTorch Profiler Blog Post `__