lightning/pytorch_lightning/core
ananthsub a18b6409d1
Check torch.distributed availability before sharded tensor state dict hook registration (#10621)
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
2021-11-19 17:34:23 +00:00
..
mixins Resolve instantiation problem with init_meta_context (#10493) 2021-11-15 19:13:01 +00:00
__init__.py Replace `yapf` with `black` (#7783) 2021-07-26 13:37:35 +02:00
datamodule.py Remove deprecated datamodule lifecycle properties (#10350) 2021-11-05 05:03:57 +00:00
decorators.py Enable auto parameters tying for TPUs (#9525) 2021-10-06 10:16:44 +02:00
hooks.py remove deprecated signature for `transfer_batch_to_device` (#10480) 2021-11-13 19:32:30 +00:00
lightning.py Check torch.distributed availability before sharded tensor state dict hook registration (#10621) 2021-11-19 17:34:23 +00:00
memory.py Deprecate LightningModule.summarize() in favor of pl.utilities.model_summary.summarize() (#8513) 2021-08-03 22:08:51 +00:00
optimizer.py Simplify `LightningOptimizer` (#10224) 2021-10-30 15:56:15 +00:00
saving.py Fix Enums parsing in generated hparms yaml (#9170) 2021-10-25 21:23:20 +00:00