no code

janeyx99 · janeyx99 · commit 38a3098fd8df · 2024-09-09T19:12:27.000Z
diff --git a/torchtune/training/_offloading.py b/torchtune/training/_offloading.py
@@ -44,12 +44,6 @@ class OffloadActivations(saved_tensors_hooks):
 
     Raises:
         ValueError: if max_fwd_stash_size is not at least 1.
-
-    Example:
-        >>> with OffloadActivations():
-        >>>     logits = model(inputs)
-        >>> loss = ...
-        >>> loss.backward()
     """
 
     def __init__(
