Proofread docstring of GPipe.worker

kakaobrain · Jun 13, 2019 · da89663 · da89663
1 parent bd1b0a8
commit da89663
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchgpipe/gpipe.py b/torchgpipe/gpipe.py
@@ -284,7 +284,7 @@ def worker(partition: Partition,
                out_queue: PriorityQueue,
                grad_enabled: bool,
                ) -> None:
-        """A worker thread runs it."""
+        """Run by worker threads."""
         torch.set_grad_enabled(grad_enabled)
 
         input: TensorOrTensors