ee9b3fe574
* Added changes for RPC plugin * Add missing kwargs * Fix code format * Loading refactors by introducing is_distributed var, fix optimizer step flow * Add rpc guard * Added docstrings and typing * resolve comments * Add additional rpc hook, refactor name of exit process hook for clarity * remove annotation * Modify behaviour to allow optional return, add test for rpc plugin * resolve tests * rename is_ddp_based * update * update for windows * update * resolve test * code smell * Revert back to init_ddp_connection for backwards compat * Swap to explicit name for property * Add missing speed parity increase for CI variability, fix call counts for child process Co-authored-by: tchaton <thomas@grid.ai> |
||
---|---|---|
.. | ||
__init__.py | ||
test_amp_plugin.py | ||
test_apex_plugin.py | ||
test_ddp_plugin.py | ||
test_plugin.py | ||
test_plugin_properties.py | ||
test_rpc_plugin.py | ||
test_sharded_plugin.py |