* fix grad norm formula
* grad-norm tracker test
* fixed seed and explicit rtol in grad norm tracking test
* a docstring for grad-norms and forced cast to float of norm_type
* support for inf-norm
* renamed the grad norm test
* docs
* fixed language in docstring
* Apply suggestions from code review
Co-authored-by: Jirka <jirka@pytorchlightning.ai>
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
* first pass for LightningModule typehints
* fix return types
* add missing types
* add type annotations to grads.py
* add type annotations to hooks.py
* add type annotation to memory.py
* proper docstring quotation marks
* add type annotations to saving.py
* fix cyclic import problem
* fix cyclic import problem
* add missing whitespace
* finish type hints for load_from_ methods
* docs: prepare_data does not return anything
* fix auto types in docs
* revert typehint for trainer in hook
* remove unnecessary return docs
* some fixes for memory docs
* revert typing for args kwargs
* added all missing None return types
* remove unused import
* add more details to dict/list return types
* fix line too long
* optimize imports
* linted
* Revert "linted"
This reverts commit 85559611e8.
* remove whitespace
* update
* update
* update
* update
* update
* changelog
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
Co-authored-by: William Falcon <waf2107@columbia.edu>
Justus Schock