diff --git a/pytorch_lightning/trainer/trainer.py b/pytorch_lightning/trainer/trainer.py index 247510df..7cc786a7 100644 --- a/pytorch_lightning/trainer/trainer.py +++ b/pytorch_lightning/trainer/trainer.py @@ -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)