Carlos Mocholí 8cc55ebdb0 Add `log_grad_norm` hook to `LightningModule` (#7873)		2021-06-08 12:09:06 +01:00
..
__init__.py	…
test_evaluation_loop.py	Add Test for memory consumption (#7733)	2021-05-28 19:40:23 +05:30
test_evaluation_loop_flow.py	Add `log_grad_norm` hook to `LightningModule` (#7873)	2021-06-08 12:09:06 +01:00
test_flow_warnings.py	…
test_training_loop.py	Move `training_output` validation to after `train_step_end` (#7868)	2021-06-08 08:37:50 +00:00
test_training_loop_flow_dict.py	…
test_training_loop_flow_scalar.py	Add `log_grad_norm` hook to `LightningModule` (#7873)	2021-06-08 12:09:06 +01:00