From 84c23d3a1d676a37f218bd8d97b1e47244d5497c Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 11 Dec 2019 06:23:21 -0800 Subject: [PATCH] remove summary --- pytorch_lightning/trainer/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)