lightning/pytorch_lightning/utilities
Eric Wiener 3f78c4ca7a
Track CPU stats with DeviceStatsMonitor (#11795)
Co-authored-by: ananthsub <ananth.subramaniam@gmail.com>
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Kaushik B <kaushikbokka@gmail.com>
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
2022-05-10 10:57:38 +00:00
..
__init__.py [1/2] Collaborative Strategy (#12842) 2022-05-05 16:06:26 +00:00
apply_func.py update apply_to_collections to support dataclass inputs (#11889) 2022-03-28 18:44:13 +02:00
argparse.py Add required for positional arguments in argparse logic (#12504) 2022-04-22 17:39:39 +02:00
auto_restart.py Improve mechanism to reset the seed after sanity check (#11870) 2022-03-01 23:27:30 +00:00
cli.py Support CLI shorthand natively (#12614) 2022-05-03 14:16:37 +02:00
cloud_io.py fix typos (#11937) 2022-02-16 17:27:51 -08:00
data.py Enforce eval shuffle warning only for default samplers (#12653) 2022-05-02 16:11:09 +00:00
deepspeed.py Update utilities API references (#11450) 2022-01-13 13:22:58 +00:00
device_parser.py Better error mesage and type checking for `gpus` arg and `devices` arg in `Trainer` (#12530) 2022-04-20 17:19:41 +00:00
distributed.py Fix false positive deprecation warning from `register_ddp_comm_hook` (#12846) 2022-04-24 23:46:18 +02:00
enums.py Add missing `f` prefix to f-strings (#12869) 2022-04-24 19:39:35 +09:00
exceptions.py fix typos (#11937) 2022-02-16 17:27:51 -08:00
fetching.py Add profiling to dataloader `next()` (#12124) 2022-05-06 11:48:12 +02: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 Track CPU stats with DeviceStatsMonitor (#11795) 2022-05-10 10:57:38 +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 Remove deprecated get_memory_profile (#12659) 2022-04-11 18:00:33 +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 Threading support for legacy loading of checkpoints (#12814) 2022-04-28 20:37:58 +00:00
model_helpers.py Update mypy (#11096) 2021-12-16 17:53:12 +01:00
model_summary.py Bump version of black to 22.3.0 (#12542) 2022-04-04 06:01:08 +02: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 Don't raise a warning when `nn.Module`s are not saved under hparams (#12669) 2022-04-08 19:40:19 +00:00
rank_zero.py general docs improvements (#12747) 2022-04-20 14:19:45 +00:00
registry.py Add `AcceleratorRegistry` (#12180) 2022-03-24 18:29:32 +00:00
seed.py Support automatic seeding of the LightningCLI (#12822) 2022-04-28 14:24:32 +00:00
signature_utils.py Add a method signature check for `setup` (#12960) 2022-05-04 18:20:59 +00:00
types.py [1/2] Collaborative Strategy (#12842) 2022-05-05 16:06:26 +00: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