lightning/pytorch_lightning/plugins
Kaushik B bf46730d92
Support TPU Pod Training (n/n) (#7296)
2021-05-17 11:33:44 +00:00
..
environments Add kubeflow cluster environment (#7300) 2021-05-17 09:05:24 +01:00
precision Use `torch.nn.utils.clip_grad_norm_` and add `clip_grad_by_value` support for TPU (#7025) 2021-05-07 16:41:39 +00:00
training_type Support TPU Pod Training (n/n) (#7296) 2021-05-17 11:33:44 +00:00
__init__.py Add Training Type Plugins Registry (#6982) 2021-04-16 18:01:56 +05:30
base_plugin.py [bugfix] Apex never instantiated. (#7274) 2021-04-30 13:16:28 -04:00
plugins_registry.py Update Accelerator Connector for Registry (#7214) 2021-05-03 21:03:21 +00:00