remove summary

This commit is contained in:
William Falcon
2019-12-11 07:27:32 -08:00
parent 486a841dab
commit 397c0754d8
-1
View File
@@ -469,7 +469,6 @@ 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']:
pass
ref_model.summarize(mode=self.weights_summary)
else:
m = "weights_summary can be None, 'full' or 'top'"