.. |
__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 a typo (#9169)
|
2021-08-28 22:56:06 +00:00 |
cli.py
|
Fix callback instantiation with CLI subcommands (#9203)
|
2021-08-30 16:44:18 +01: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 calls to internal dev debugger in training- and eval loop (#9188)
|
2021-08-30 17:16:59 +02:00 |
deepspeed.py
|
Add functions to collate deepspeed zero 3 checkpoints (#8701)
|
2021-08-04 09:39:02 +00:00 |
device_dtype_mixin.py
|
…
|
|
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
|
…
|
|
fetching.py
|
3/n inter batch parallelism (#9052)
|
2021-08-24 18:45:54 +00:00 |
finite_checks.py
|
…
|
|
grads.py
|
…
|
|
imports.py
|
Add support for CPU AMP autocast (#9084)
|
2021-08-25 12:18:00 +00:00 |
memory.py
|
Fix mypy for `utilities.memory` (#8744)
|
2021-08-06 11:21:09 +02:00 |
metrics.py
|
Fix comments for metrics_to_scalars (#8782)
|
2021-08-07 15:33:36 +05:30 |
model_helpers.py
|
…
|
|
model_summary.py
|
feat: Add Rich Progress Bar (#8929)
|
2021-08-24 02:40:36 +00:00 |
parsing.py
|
…
|
|
seed.py
|
…
|
|
signature_utils.py
|
3/n inter batch parallelism (#9052)
|
2021-08-24 18:45:54 +00:00 |
types.py
|
Introduce CheckpointIO Plugin (#8743)
|
2021-08-13 17:35:31 +01:00 |
upgrade_checkpoint.py
|
Automatic string fixes (#8886)
|
2021-08-13 14:28:14 +00:00 |
warnings.py
|
…
|
|
xla_device.py
|
Fix mypy for `utilities.xla_device` (#8755)
|
2021-08-06 17:01:21 +02:00 |