b29757da90
* Implemented ModelSummary total params values Signed-off-by: George Corrêa de Araújo <george.gcac@gmail.com> * Fixed documentation, handling modules that are containers for other modules when calculating total params Signed-off-by: gca <george.gcac@gmail.com> * Reduced max line length, updated total number of params layout Signed-off-by: gca <george.gcac@gmail.com> * Now using only top-level modules of main module to calculate total params Signed-off-by: gca <george.gcac@gmail.com> * Added default value for named_modules param in summarize function Signed-off-by: gca <george.gcac@gmail.com> * Removed summary function params, removed unused properties Signed-off-by: gca <george.gcac@gmail.com> * Changed from np.prod(shape) to numel Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com> * changelog * Update pytorch_lightning/core/memory.py Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com> Co-authored-by: chaton <thomas@grid.ai> Co-authored-by: Nicki Skafte <skaftenicki@gmail.com> Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com> Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com> |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| datamodule.py | ||
| decorators.py | ||
| grads.py | ||
| hooks.py | ||
| lightning.py | ||
| memory.py | ||
| saving.py | ||
| step_result.py | ||