Skip to content

Commit 8d6e7dc

Browse files
committed
removes loss fn unused
1 parent 3c86d8f commit 8d6e7dc

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

recipes_source/distributed_checkpoint_recipe.rst

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -249,7 +249,6 @@ The reason that we need the ``state_dict`` prior to loading is:
249249
model = ToyModel().to(rank)
250250
model = FSDP(model)
251251
252-
loss_fn = nn.MSELoss()
253252
optimizer = torch.optim.Adam(model.parameters(), lr=0.1)
254253
255254
state_dict = { "app": AppState(model, optimizer)}

0 commit comments

Comments
 (0)