We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent b93e062 commit c2a6648Copy full SHA for c2a6648
checkpoint_engine/distributed/base.py
@@ -45,7 +45,7 @@ def all_gather_object(
45
def all_reduce(
46
self,
47
tensor: torch.Tensor,
48
- op :torch_dist.ReduceOp,
+ op: torch_dist.ReduceOp,
49
group,
50
):
51
raise NotImplementedError
0 commit comments