pytorch
0c46b6b3 - [DDP] Enhance warning for find_unused_params (#52385)

Commit
4 years ago
[DDP] Enhance warning for find_unused_params (#52385) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/52385 This warning should specify that we did not find unused params in the _forward_ pass, which is when we log this warning. This is to avoid confusion when we get an error because not all outputs were used to compute loss, which also raises an error about unused parameters (to be fixed in the next diff) ghstack-source-id: 122001929 Test Plan: CI Reviewed By: zhaojuanmao Differential Revision: D26494136 fbshipit-source-id: d9b41732ea7e5e31b899d590d311080e3dc56682
Author
Parents
Loading