mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-30 13:40:46 +08:00
remove summary
This commit is contained in:
1 parent
e8fb2fc111
commit
84c23d3a1d
1 file changed
+1
-1
@@ -470,7 +470,7 @@ class Trainer(TrainerIOMixin,
|
||||
if self.proc_rank == 0 and self.weights_summary is not None:
|
||||
if self.weights_summary in ['full', 'top']:
|
||||
pass
|
||||
# ref_model.summarize(mode=self.weights_summary)
|
||||
ref_model.summarize(mode=self.weights_summary)
|
||||
else:
|
||||
m = "weights_summary can be None, 'full' or 'top'"
|
||||
raise MisconfigurationException(m)
|
||||
|
||||
Reference in new issue
Block a user