10005355922be53768b03c8199674e37f9ece772,dataset/models/tf/squeezenet.py,SqueezeNet,build_config,#SqueezeNet#Any#,50

Before Change

    def build_config(self, names=None):
        config = super().build_config(names)
        config["head"]["filters"] = self.num_classes("targets")
        return config

    @classmethod
    def body(cls, inputs, name="body", **kwargs):

After Change

        return config

    def build_config(self, names=None):
        config = super().build_config(names)
        if config.get("head/units") is None:
            config["head/units"] = self.num_classes("targets")
        if config.get("head/filters") is None:
            config["head/filters"] = self.num_classes("targets")
        return config

    @classmethod
    def body(cls, inputs, name="body", **kwargs):

Instances