remove summary

2026-09-11 12:31:23 +08:00 · 2019-12-11 06:23:21 -08:00
parent e8fb2fc111
commit 84c23d3a1d
1 changed files with 1 additions and 1 deletions
@@ -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)