| .. |
|
__init__.py
|
…
|
|
|
ddp_model.py
|
Update `gpus` flag with `accelerator` and `devices` flag (#12156)
|
2022-03-23 19:52:12 +00:00 |
|
test_bagua_strategy.py
|
Drop PyTorch 1.7 support (#12432)
|
2022-03-27 21:31:20 +00:00 |
|
test_common.py
|
Support `strategy` argument being case insensitive (#12528)
|
2022-04-13 14:14:29 -04:00 |
|
test_custom_plugin.py
|
Add `SyncBatchNormPlugin` (#11754)
|
2022-03-01 19:41:40 +05:30 |
|
test_ddp.py
|
Update tests in `strategies` directory in preparation for #11040 (#12467)
|
2022-03-27 18:13:48 +02:00 |
|
test_ddp_fully_sharded_with_full_state_dict.py
|
Update standalone tests (#12472)
|
2022-03-28 04:16:17 +02:00 |
|
test_ddp_spawn.py
|
Update `tests/accelerators/*.py` to use `devices` instead of `gpus` or `ipus` (#11817)
|
2022-03-24 14:09:39 +00:00 |
|
test_ddp_spawn_strategy.py
|
Update Plugins doc (#12440)
|
2022-03-29 12:09:41 +00:00 |
|
test_ddp_strategy.py
|
Update Plugins doc (#12440)
|
2022-03-29 12:09:41 +00:00 |
|
test_ddp_strategy_with_comm_hook.py
|
Update standalone tests (#12472)
|
2022-03-28 04:16:17 +02:00 |
|
test_deepspeed_strategy.py
|
Update standalone tests (#12472)
|
2022-03-28 04:16:17 +02:00 |
|
test_dp.py
|
Update tests in `strategies` directory in preparation for #11040 (#12467)
|
2022-03-27 18:13:48 +02:00 |
|
test_sharded_strategy.py
|
Update standalone tests (#12472)
|
2022-03-28 04:16:17 +02:00 |
|
test_single_device_strategy.py
|
Drop PyTorch 1.7 support (#12432)
|
2022-03-27 21:31:20 +00:00 |
|
test_strategy_registry.py
|
…
|
|
|
test_tpu_spawn.py
|
Update tpu_cores flag with accelerator and devices flag (#12158)
|
2022-03-25 11:57:02 +00:00 |
7122a597ed