Skip to content

GitLab

Unverified Commit 2f204bca authored by mcarilli's avatar mcarilli Committed by GitHub
Browse files

fix error message

parent 89fa152b
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
apex/parallel/distributed.py
View file @ 2f204bca
...@@ -318,7 +318,7 @@ class DistributedDataParallel(Module): ...@@ -318,7 +318,7 @@ class DistributedDataParallel(Module):
self.ready_buckets_not_reduced.remove(i) self.ready_buckets_not_reduced.remove(i)
self.next_bucket += 1 self.next_bucket += 1
else: else:
raise ValueError("i should always be <= next_bucket") raise ValueError("i should always be >= next_bucket")
else: else:
self.ready_buckets_not_reduced.add(bucket_idx) self.ready_buckets_not_reduced.add(bucket_idx)
......
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