Skip to content

Commit 49253dd

Browse files
committed
Add note about default_auto_wrap_policy -> size_based_auto_wrap_policy
1 parent bbe7afe commit 49253dd

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

intermediate_source/FSDP_tutorial.rst

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -62,6 +62,9 @@ We add the following code snippets to a python script “FSDP_mnist.py”.
6262

6363
1.2 Import necessary packages
6464

65+
.. note::
66+
This tutorial is intended for pytorch versions 1.12 and later. If you are using 1.11 or earlier, replace all instances of `size_based_auto_wrap_policy` with `default_auto_wrap_policy`.
67+
6568
.. code-block:: python
6669
6770
# Based on: https://github.com/pytorch/examples/blob/master/mnist/main.py

0 commit comments

Comments
 (0)