diff options
| -rw-r--r-- | torch/nn/parallel/data_parallel.py | 2 | ||||
| -rw-r--r-- | torch/nn/parallel/distributed.py | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/torch/nn/parallel/data_parallel.py b/torch/nn/parallel/data_parallel.py index 2f43d49ea2..817015cb87 100644 --- a/torch/nn/parallel/data_parallel.py +++ b/torch/nn/parallel/data_parallel.py @@ -27,7 +27,7 @@ class DataParallel(Module): the model's forward pass. .. warning:: - Forward and backwrad hooks defined on :attr:`module` and its submodules + Forward and backward hooks defined on :attr:`module` and its submodules won't be invoked anymore, unless the hooks are initialized in the :meth:`forward` method. diff --git a/torch/nn/parallel/distributed.py b/torch/nn/parallel/distributed.py index 3fb1ffffcf..b57c086615 100644 --- a/torch/nn/parallel/distributed.py +++ b/torch/nn/parallel/distributed.py @@ -81,7 +81,7 @@ class DistributedDataParallel(Module): 0, to all other replicas in the system in every iteration. .. warning:: - Forward and backwrad hooks defined on :attr:`module` and its submodules + Forward and backward hooks defined on :attr:`module` and its submodules won't be invoked anymore, unless the hooks are initialized in the :meth:`forward` method. |