We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent a62458d commit a2f528dCopy full SHA for a2f528d
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