Skip to content

GitLab

Commit b7f10ad0 authored by Michael Carilli's avatar Michael Carilli
Browse files

typo

parent 8521bb22
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
apex/parallel/distributed.py
View file @ b7f10ad0
...@@ -411,7 +411,7 @@ class DistributedDataParallel(Module): ...@@ -411,7 +411,7 @@ class DistributedDataParallel(Module):
return self.bucket_events[0] return self.bucket_events[0]
def allreduce_bucket(self, bucket): def allreduce_bucket(self, bucket, bucket_idx):
tensor = flatten(bucket) tensor = flatten(bucket)
tensor_to_allreduce = tensor tensor_to_allreduce = tensor
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment