Kaushik B 1b3e4f9fb9 Fix sync_dist for tpus (#6950)		2021-04-13 14:17:15 +05:30
..
__init__.py	CI: add flake8 (#4239)	2020-10-19 21:20:17 +01:00
datamodule.py	Support teardown hook on DataModule (#4673)	2021-03-25 07:51:55 -05:00
decorators.py	Add before_batch_transfer and after_batch_transfer hooks (#3671)	2021-02-18 06:58:12 -05:00
grads.py	formatting 5/n: Core (#5721)	2021-02-08 14:29:43 -05:00
hooks.py	[Fix] Ensure we set the eval/train flag correctly on accelerator model (#6877)	2021-04-08 14:04:26 -04:00
lightning.py	[RFC] Add `self.lr_schedulers()` to LightningModule for manual optimization (#6567)	2021-04-09 10:32:14 +01:00
memory.py	Handle torch.jit scripted modules in layer summary (#6511)	2021-03-15 03:17:42 +01:00
optimizer.py	[bugfix] Check LightningOptimizer doesn't delete optimizer hooks (#6305)	2021-03-04 20:11:59 +00:00
saving.py	Fix csv extension check (#6436)	2021-04-08 01:16:31 +00:00
step_result.py	Fix sync_dist for tpus (#6950)	2021-04-13 14:17:15 +05:30