remove summary

2026-09-09 11:32:07 +08:00 · 2019-12-11 06:21:36 -08:00
parent d1633aac11
commit e8fb2fc111
1 changed files with 2 additions and 1 deletions
@@ -469,7 +469,8 @@ class Trainer(TrainerIOMixin,
        # print model summary
        if self.proc_rank == 0 and self.weights_summary is not None:
            if self.weights_summary in ['full', 'top']:
-                ref_model.summarize(mode=self.weights_summary)
+                pass
+                # ref_model.summarize(mode=self.weights_summary)
            else:
                m = "weights_summary can be None, 'full' or 'top'"
                raise MisconfigurationException(m)