| .. |
|
__init__.py
|
Drop PyTorch 1.7 support (#12432)
|
2022-03-27 21:31:20 +00:00 |
|
apply_func.py
|
update apply_to_collections to support dataclass inputs (#11889)
|
2022-03-28 18:44:13 +02:00 |
|
argparse.py
|
Replace `eval()` with `ast.literal_eval()` for security (#12212)
|
2022-03-05 18:38:16 +00:00 |
|
auto_restart.py
|
Improve mechanism to reset the seed after sanity check (#11870)
|
2022-03-01 23:27:30 +00:00 |
|
cli.py
|
Add `LightningCLI(auto_registry)` (#12108)
|
2022-03-08 12:26:10 -05:00 |
|
cloud_io.py
|
fix typos (#11937)
|
2022-02-16 17:27:51 -08:00 |
|
data.py
|
Do not prefetch when possible (#12101)
|
2022-02-28 18:31:18 +00:00 |
|
deepspeed.py
|
Update utilities API references (#11450)
|
2022-01-13 13:22:58 +00:00 |
|
device_parser.py
|
Add cpu device parser to validate cpu devices (#12160)
|
2022-03-28 20:33:04 +05:30 |
|
distributed.py
|
Drop PyTorch 1.7 support (#12432)
|
2022-03-27 21:31:20 +00:00 |
|
enums.py
|
Add support for Habana accelerator (HPU) (#11808)
|
2022-03-25 10:24:52 +00:00 |
|
exceptions.py
|
fix typos (#11937)
|
2022-02-16 17:27:51 -08:00 |
|
fetching.py
|
Do not prefetch when possible (#12101)
|
2022-02-28 18:31:18 +00:00 |
|
finite_checks.py
|
Fix logging of nan parameters (#9364)
|
2021-09-09 00:39:23 +00:00 |
|
grads.py
|
Group all the logged gradients under the same sub-folder (#7756)
|
2021-10-20 15:48:36 +00:00 |
|
imports.py
|
Fix _module_available to detect horovod.torch properly (#12377)
|
2022-03-28 16:00:45 +00:00 |
|
logger.py
|
Refactor codebase to use `trainer.loggers` over `trainer.logger` when needed (#11920)
|
2022-02-25 16:01:04 -08:00 |
|
memory.py
|
Add typing for utilities/memory.py (#11545)
|
2022-02-03 02:34:05 +00:00 |
|
meta.py
|
Add `LightningCLI(auto_registry)` (#12108)
|
2022-03-08 12:26:10 -05:00 |
|
metrics.py
|
Add typing to some utility files (#11316)
|
2022-01-05 17:14:22 +00:00 |
|
migration.py
|
Add typing to some utility files (#11316)
|
2022-01-05 17:14:22 +00:00 |
|
model_helpers.py
|
Update mypy (#11096)
|
2021-12-16 17:53:12 +01:00 |
|
model_summary.py
|
Drop PyTorch 1.7 support (#12432)
|
2022-03-27 21:31:20 +00:00 |
|
optimizer.py
|
Refactor `Strategy._move_optimizer_states` as utility functions (#11758)
|
2022-02-18 08:36:07 +00:00 |
|
parameter_tying.py
|
Configure mypy to install dependencies in CI and update pyproject.toml (#10682)
|
2021-11-25 00:19:30 +00:00 |
|
parsing.py
|
Add cpu device parser to validate cpu devices (#12160)
|
2022-03-28 20:33:04 +05:30 |
|
rank_zero.py
|
Fix environment variable order for global rank determination (#11406)
|
2022-02-17 02:09:05 +00:00 |
|
registry.py
|
Add `AcceleratorRegistry` (#12180)
|
2022-03-24 18:29:32 +00:00 |
|
seed.py
|
Improve mechanism to reset the seed after sanity check (#11870)
|
2022-03-01 23:27:30 +00:00 |
|
signature_utils.py
|
3/n inter batch parallelism (#9052)
|
2021-08-24 18:45:54 +00:00 |
|
types.py
|
Rename `_SupportsStateDict` --> `_Stateful` Protocol (#11469)
|
2022-02-02 23:45:59 +01:00 |
|
upgrade_checkpoint.py
|
Add typing to some utility files (#11316)
|
2022-01-05 17:14:22 +00:00 |
|
warnings.py
|
Centralize rank_zero_only utilities into their own module (#11747)
|
2022-02-07 08:09:55 +00:00 |
|
xla_device.py
|
guard against None in pytorch get_xla_supported_devices (#9572)
|
2021-10-12 12:01:32 +00:00 |
834115b859