| .. |
|
launchers
|
Fix `is_interactive_compatible` logic after AcceleratorConnector rewrite (#12008)
|
2022-02-22 19:20:54 +05:30 |
|
__init__.py
|
[1/2] Collaborative Strategy (#12842)
|
2022-05-05 16:06:26 +00:00 |
|
bagua.py
|
fix typing in BaguaStrategy (#12692)
|
2022-04-11 09:58:00 +02:00 |
|
collaborative.py
|
[1/2] Collaborative Strategy (#12842)
|
2022-05-05 16:06:26 +00:00 |
|
ddp.py
|
Enable all ddp params for hpu parallel strategy (#13067)
|
2022-05-19 12:41:16 +05:30 |
|
ddp2.py
|
Rewrite accelerator_connector (#11448)
|
2022-02-17 23:38:39 +00:00 |
|
ddp_spawn.py
|
Fix double precision during evaluation (#12983)
|
2022-05-11 17:43:19 +00:00 |
|
deepspeed.py
|
Fix default int values being float (#12989)
|
2022-05-06 10:24:19 +00:00 |
|
dp.py
|
Rewrite accelerator_connector (#11448)
|
2022-02-17 23:38:39 +00:00 |
|
fully_sharded.py
|
Fix default int values being float (#12989)
|
2022-05-06 10:24:19 +00:00 |
|
fully_sharded_native.py
|
[FSDP] Adding Native FSDP Strategy (#12447)
|
2022-05-05 12:48:29 +00: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
|
Enable all ddp params for hpu parallel strategy (#13067)
|
2022-05-19 12:41:16 +05:30 |
|
ipu.py
|
Override `optimizer_zero_grad` when using the `IPUStrategy` (#12913)
|
2022-05-01 22:47:21 +00: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
|
Break lazy accumulation of graphs (#12938)
|
2022-05-05 21:31:41 +02:00 |
|
single_tpu.py
|
Rewrite accelerator_connector (#11448)
|
2022-02-17 23:38:39 +00:00 |
|
strategy.py
|
Update core/lightning.py to core/module.py (#12740)
|
2022-05-18 08:03:55 +05:30 |
|
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 |
74941bb790