8283680aa0
* Allow ddp plugin to move the input to a different device if needed * Swapped name to on_before_forward to align with hooks in the future * Update pytorch_lightning/plugins/ddp_plugin.py Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com> * Pass variable arg type to hook, add example * Remove blank space (pep check) * Added blank line Co-authored-by: William Falcon <waf2107@columbia.edu> Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com> |
||
|---|---|---|
| .. | ||
| accelerators | ||
| callbacks | ||
| cluster_environments | ||
| core | ||
| distributed | ||
| loggers | ||
| metrics | ||
| overrides | ||
| plugins | ||
| profiler | ||
| trainer | ||
| tuner | ||
| utilities | ||
| __init__.py | ||
| py.typed | ||