Profiler - Function time profiler

sizhky · Oct 1, 2024 · 974a5a2 · 974a5a2
1 parent 37107b1
commit 974a5a2
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/nbs/profiler.ipynb b/nbs/profiler.ipynb
@@ -15,7 +15,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "# | default_exp misc"
+    "# | default_exp profiler"
    ]
   },
   {

diff --git a/torch_snippets/profiler.py b/torch_snippets/profiler.py
@@ -1,8 +1,14 @@
+# AUTOGENERATED! DO NOT EDIT! File to edit: ../nbs/profiler.ipynb.
+
+# %% auto 0
+__all__ = ['time_profiler']
+
+# %% ../nbs/profiler.ipynb 2
 import cProfile
 import pstats
 from io import StringIO
 
-
+# %% ../nbs/profiler.ipynb 3
 def time_profiler(filename='profiling_results.txt'):
     # The outer function that allows customization of the filename
     def decorator(func):
@@ -22,4 +28,4 @@ def wrapper(*args, **kwargs):
                 f.write(s.getvalue())
             return result
         return wrapper
-    return decorator
+    return decorator