| .. |
|
launchers
|
Fix `is_interactive_compatible` logic after AcceleratorConnector rewrite (#12008)
|
2022-02-22 19:20:54 +05:30 |
|
__init__.py
|
Add support for Habana accelerator (HPU) (#11808)
|
2022-03-25 10:24:52 +00:00 |
|
bagua.py
|
fix typing in BaguaStrategy (#12692)
|
2022-04-11 09:58:00 +02:00 |
|
ddp.py
|
Fix import error when `distributed` module not available (#12794)
|
2022-04-20 17:28:39 +00:00 |
|
ddp2.py
|
Rewrite accelerator_connector (#11448)
|
2022-02-17 23:38:39 +00:00 |
|
ddp_spawn.py
|
Drop PyTorch 1.7 support (#12432)
|
2022-03-27 21:31:20 +00:00 |
|
deepspeed.py
|
Fix to ensure the checkpoint states are saved in a common filepath with deepspeed (#12887)
|
2022-04-27 14:41:51 +02:00 |
|
dp.py
|
Rewrite accelerator_connector (#11448)
|
2022-02-17 23:38:39 +00:00 |
|
fully_sharded.py
|
Reduce number of times optimizers are instantiated with FSDP (#12267)
|
2022-03-21 18:18:59 +01:00 |
|
horovod.py
|
Support gradient accumulation using Horovod's `backward_passes_per_step` (#11911)
|
2022-02-19 02:54:04 +01:00 |
|
hpu_parallel.py
|
Add support for Habana accelerator (HPU) (#11808)
|
2022-03-25 10:24:52 +00:00 |
|
ipu.py
|
Remove deprecated `dataloader_idx` argument from `on_train_batch_start/end` callback hooks (#12769)
|
2022-04-18 18:49:59 +02:00 |
|
parallel.py
|
Add support for specifying process group backend to relevant distributed strategies (#11745)
|
2022-03-17 23:38:03 -07:00 |
|
sharded.py
|
Bump version of black to 22.3.0 (#12542)
|
2022-04-04 06:01:08 +02:00 |
|
sharded_spawn.py
|
Rewrite accelerator_connector (#11448)
|
2022-02-17 23:38:39 +00:00 |
|
single_device.py
|
Rewrite accelerator_connector (#11448)
|
2022-02-17 23:38:39 +00:00 |
|
single_hpu.py
|
Add support for Habana accelerator (HPU) (#11808)
|
2022-03-25 10:24:52 +00:00 |
|
single_tpu.py
|
Rewrite accelerator_connector (#11448)
|
2022-02-17 23:38:39 +00:00 |
|
strategy.py
|
Remove deprecated `dataloader_idx` argument from `on_train_batch_start/end` callback hooks (#12769)
|
2022-04-18 18:49:59 +02:00 |
|
strategy_registry.py
|
Add `AcceleratorRegistry` (#12180)
|
2022-03-24 18:29:32 +00:00 |
|
tpu_spawn.py
|
Support passing `storage_options` in `trainer.save_checkpoint()` API (#11891)
|
2022-03-09 18:35:50 +00:00 |
|
utils.py
|
Introduce strategies directory for Training Strategies (#11226)
|
2021-12-22 20:23:30 +00:00 |
70754bea83