lightning/pytorch_lightning/utilities
Sean Naren 72bb0186fb
Update requirements, update test (#9345)
2021-09-06 12:58:54 +01:00
..
__init__.py Add support for CPU AMP autocast (#9084) 2021-08-25 12:18:00 +00:00
apply_func.py Make device and dtype required (#9168) 2021-08-28 23:13:05 +00:00
argparse.py Add bfloat16 support to Lightning Trainer (#9049) 2021-08-24 09:47:21 +00:00
argparse_utils.py Update removal version of argparse_utils.py from 1.4 to 2.0 for backwards compatibility (#9162) 2021-08-27 14:10:28 +02:00
auto_restart.py fix state extraction from batch when fault-tolerant training (#9281) 2021-09-02 11:57:40 -07:00
cli.py Allow easy CLI trainer re-instantiation (#9241) 2021-09-03 00:56:30 +02:00
cloud_io.py Fix mypy typing for `utilities.cloud_io.py` (#8671) 2021-08-03 11:56:28 +02:00
data.py Add tests for functions in utilities/data.py (#8785) 2021-08-10 07:39:00 +01:00
debugging.py remove early stopping tracking from internal debugger (#9327) 2021-09-06 10:43:03 +02:00
deepspeed.py Minor improvement to deepspeed utility comments (#9224) 2021-08-31 18:19:59 +02:00
device_dtype_mixin.py CI: yesqa (#8564) 2021-08-02 16:05:56 +00:00
device_parser.py remove deprecation of gpu string parsing behavior (#8770) 2021-08-06 15:41:03 +00:00
distributed.py Fix `self.log(sync_dist=True, reduce_fx={mean,max})` (#9142) 2021-08-27 15:40:51 +00:00
enums.py Add validate logic for precision (#9080) 2021-08-24 20:00:09 +00:00
exceptions.py [bugfix] Add mechanism to prevent deadlock for DDP on Exception Trigger (#8167) 2021-06-28 19:26:03 +00:00
fetching.py 3/n inter batch parallelism (#9052) 2021-08-24 18:45:54 +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 requirements, update test (#9345) 2021-09-06 12:58:54 +01:00
memory.py Move tracking epoch end outputs logic to the `EvaluationEpochLoop` (#9261) 2021-09-03 15:02:34 +02:00
metrics.py Fix comments for metrics_to_scalars (#8782) 2021-08-07 15:33:36 +05:30
model_helpers.py Deprecate on_keyboard_interrupt callback hook (#9260) 2021-09-06 09:57:00 +02:00
model_summary.py feat: Add Rich Progress Bar (#8929) 2021-08-24 02:40:36 +00:00
parsing.py Fix inspection of unspecified args for container hparams (#9125) 2021-09-06 09:48:11 +00:00
seed.py Replace `yapf` with `black` (#7783) 2021-07-26 13:37:35 +02:00
signature_utils.py 3/n inter batch parallelism (#9052) 2021-08-24 18:45:54 +00:00
types.py Add typing for `ResultCollection` [3/3] (#9271) 2021-09-06 09:34:40 +00:00
upgrade_checkpoint.py Automatic string fixes (#8886) 2021-08-13 14:28:14 +00:00
warnings.py Replace `yapf` with `black` (#7783) 2021-07-26 13:37:35 +02:00
xla_device.py Fix mypy for `utilities.xla_device` (#8755) 2021-08-06 17:01:21 +02:00