It seems that pruner.step(interactive=True) returns nothing

Open ArthurRyan0803 opened this issue 2 years ago • 1 comments

I want to use interactive pruning, but the step method of GroupNormRruner returns nothing when interactive = True.

    def step(self, interactive=False): 
        super(GroupNormPruner, self).step(interactive=interactive)
        # update the group list after pruning
        self._groups = list(self.DG.get_all_groups(root_module_types=self.root_module_types, ignored_layers=self.ignored_layers))

Dec 12 '23 08:12 ArthurRyan0803

Oh! you are right! There is a bug in this function. Will fix it soon.

Dec 12 '23 08:12 VainF