HHousen / lecture2notes

Mismatched parameters in overridden method PYL-W0221
Bug risk
Major
3 years ago4 years old
Parameters differ from overridden 'generate' method
139            return pool[target_layer]
140        raise ValueError("Invalid layer name: {}".format(target_layer))
141
142    def generate(self, target_layer):143        fmaps = self._find(self.fmap_pool, target_layer)
144        grads = self._find(self.grad_pool, target_layer)
145        weights = F.adaptive_avg_pool2d(grads, 1)