Skip to content

Commit 5b68901

Browse files
committed
another indentation fix
1 parent 51f640e commit 5b68901

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

intermediate_source/FSDP_tutorial.rst

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -217,8 +217,8 @@ We add the following code snippets to a python script “FSDP_mnist.py”.
217217
train_loader = torch.utils.data.DataLoader(dataset1,**train_kwargs)
218218
test_loader = torch.utils.data.DataLoader(dataset2, **test_kwargs)
219219
my_auto_wrap_policy = functools.partial(
220-
size_based_auto_wrap_policy, min_num_params=100
221-
)
220+
size_based_auto_wrap_policy, min_num_params=100
221+
)
222222
torch.cuda.set_device(rank)
223223
224224

0 commit comments

Comments
 (0)