Skip to content

Commit 0e52e07

Browse files
committed
Fix missing import of Stateful (#3067)
1 parent d490aef commit 0e52e07

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

recipes_source/distributed_checkpoint_recipe.rst

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -178,6 +178,7 @@ The reason that we need the ``state_dict`` prior to loading is:
178178
import torch
179179
import torch.distributed as dist
180180
import torch.distributed.checkpoint as dcp
181+
from torch.distributed.checkpoint.stateful import Stateful
181182
from torch.distributed.checkpoint.state_dict import get_state_dict, set_state_dict
182183
import torch.multiprocessing as mp
183184
import torch.nn as nn

0 commit comments

Comments
 (0)