lightning/pytorch_lightning/utilities
Sean Naren 560a5c3fc5
Add functions to collate deepspeed zero 3 checkpoints (#8701)
2021-08-04 09:39:02 +00:00
..
__init__.py Prune deprecated metrics (#8586) 2021-07-28 16:57:31 +00:00
apply_func.py Replace `yapf` with `black` (#7783) 2021-07-26 13:37:35 +02:00
argparse.py Fix mypy in utilities.argparse (#8124) 2021-07-30 16:36:55 +00:00
argparse_utils.py CI: yesqa (#8564) 2021-08-02 16:05:56 +00:00
auto_restart.py update logic to inject FastForwardSampler / CaptureIterableDataset 2/n (#8366) 2021-08-02 20:52:06 +00:00
cli.py Prune deprecated metrics (#8586) 2021-07-28 16:57:31 +00:00
cloud_io.py Fix mypy typing for `utilities.cloud_io.py` (#8671) 2021-08-03 11:56:28 +02:00
data.py Replace `yapf` with `black` (#7783) 2021-07-26 13:37:35 +02:00
debugging.py Avoid unnecessary list creation (#8595) 2021-07-28 13:36:45 +05:30
deepspeed.py Add functions to collate deepspeed zero 3 checkpoints (#8701) 2021-08-04 09:39:02 +00:00
device_dtype_mixin.py CI: yesqa (#8564) 2021-08-02 16:05:56 +00:00
device_parser.py Add check for unique device ids (#8666) 2021-08-03 08:18:51 +00:00
distributed.py Replace `yapf` with `black` (#7783) 2021-07-26 13:37:35 +02:00
enums.py Add ddp_cpu to DistributedType Enum (#8596) 2021-07-29 10:02:32 +02:00
exceptions.py [bugfix] Add mechanism to prevent deadlock for DDP on Exception Trigger (#8167) 2021-06-28 19:26:03 +00:00
finite_checks.py Replace `yapf` with `black` (#7783) 2021-07-26 13:37:35 +02:00
grads.py Replace `yapf` with `black` (#7783) 2021-07-26 13:37:35 +02:00
imports.py update logic to inject FastForwardSampler / CaptureIterableDataset 2/n (#8366) 2021-08-02 20:52:06 +00:00
memory.py Deprecate LightningModule.summarize() in favor of pl.utilities.model_summary.summarize() (#8513) 2021-08-03 22:08:51 +00:00
metrics.py Replace `yapf` with `black` (#7783) 2021-07-26 13:37:35 +02:00
model_helpers.py Replace `yapf` with `black` (#7783) 2021-07-26 13:37:35 +02:00
model_summary.py Deprecate LightningModule.summarize() in favor of pl.utilities.model_summary.summarize() (#8513) 2021-08-03 22:08:51 +00:00
parsing.py Docs improvements around hparams (#8577) 2021-07-30 11:06:03 +00:00
seed.py Replace `yapf` with `black` (#7783) 2021-07-26 13:37:35 +02:00
signature_utils.py Add `dataloader_idx` to batch transfer hooks (#6241) 2021-05-13 23:03:55 +05:30
types.py Replace `yapf` with `black` (#7783) 2021-07-26 13:37:35 +02:00
upgrade_checkpoint.py fix duplicate console logging bug v2 (#6275) 2021-03-02 15:17:55 +05:30
warnings.py Replace `yapf` with `black` (#7783) 2021-07-26 13:37:35 +02:00
xla_device.py Replace `yapf` with `black` (#7783) 2021-07-26 13:37:35 +02:00