wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-09 11:32:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

Param printing (#336)

Browse Source 
* print thousands as K, M, B, T, ...

* add option to print top-level modules only

* added doc string and added spacing

* do not print summary if neither "full" nor "top"

* updated docs showing summary print options

* fix line length for travis
This commit is contained in:
Adrian Wälchli
2019-10-08 15:30:06 -04:00
committed by William Falcon
parent ff2a21a08a
commit 6e3e740a7f
4 changed files with 63 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/root_module/root_module.py
+2 -2
View File
@@ -159,8 +159,8 @@ class LightningModule(GradInformation, ModelIO, ModelHooks):
return model
def summarize(self):
model_summary = ModelSummary(self)
def summarize(self, mode):
model_summary = ModelSummary(self, mode=mode)
print(model_summary)
def freeze(self):