db784225eb
* simplified model size calc * fix spaces * fix newlines * minor refactor * Update pytorch_lightning/core/memory.py Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com> * make model size property * fix doctest * Update pytorch_lightning/core/memory.py Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com> * remove explicit doctest from file * better docs * model precalculate size 1.0 mbs * better comment * Update tests/core/test_memory.py Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com> * Update tests/core/test_memory.py Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com> * merge _model_size into model_size property itself * minor comment fix * add feature to changelog * added precision test * isort * minor def name typo * remove monkeypath set env as boringmodel wont need any torch hub cache Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com> Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com> Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com> |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| test_datamodules.py | ||
| test_decorators.py | ||
| test_lightning_module.py | ||
| test_lightning_optimizer.py | ||
| test_memory.py | ||
| test_metric_result_integration.py | ||
| test_results.py | ||