| .. |
|
__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
|
Make standalone tests less verbose (#12684)
|
2022-04-20 20:57:40 +02:00 |
|
test_collaborative.py
|
[1/2] Collaborative Strategy (#12842)
|
2022-05-05 16:06:26 +00:00 |
|
test_common.py
|
Support `strategy` argument being case insensitive (#12528)
|
2022-03-30 14:19:19 +00:00 |
|
test_custom_plugin.py
|
Add `SyncBatchNormPlugin` (#11754)
|
2022-03-01 19:41:40 +05:30 |
|
test_ddp.py
|
Make standalone tests less verbose (#12684)
|
2022-04-20 20:57:40 +02:00 |
|
test_ddp_fully_sharded_native.py
|
[FSDP] Adding Native FSDP Strategy (#12447)
|
2022-05-05 12:48:29 +00:00 |
|
test_ddp_fully_sharded_with_full_state_dict.py
|
[FIX] Enable mixed precision in the Fully Sharded Strategy when `precision=16` (#12965)
|
2022-05-03 15:39:59 +00:00 |
|
test_ddp_spawn.py
|
Remove deprecated get_memory_profile (#12659)
|
2022-04-11 18:00:33 +00:00 |
|
test_ddp_spawn_strategy.py
|
Update Plugins doc (#12440)
|
2022-03-29 12:09:41 +00:00 |
|
test_ddp_strategy.py
|
Make standalone tests less verbose (#12684)
|
2022-04-20 20:57:40 +02:00 |
|
test_ddp_strategy_with_comm_hook.py
|
Fix tests related to DDP communication hooks (#12878)
|
2022-04-27 22:37:19 +05:30 |
|
test_deepspeed_strategy.py
|
Raise an exception when using DeepSpeed with an invalid accelerator (#12699)
|
2022-05-04 18:16:41 +00:00 |
|
test_dp.py
|
Remove deprecated get_memory_profile (#12659)
|
2022-04-11 18:00:33 +00:00 |
|
test_sharded_strategy.py
|
Make standalone tests less verbose (#12684)
|
2022-04-20 20:57:40 +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
|
Rewrite accelerator_connector (#11448)
|
2022-02-17 23:38:39 +00:00 |
|
test_tpu_spawn.py
|
Update tpu_cores flag with accelerator and devices flag (#12158)
|
2022-03-25 11:57:02 +00:00 |
1a502c061c