Method get_z_from_zs

TinyCLIP/src/open_clip/l0module.py:256–264

(self, zs)

        return soft_mask

    def get_z_from_zs(self, zs):
        numpified_zs = {}
        # for t in self.all_types:
        #     name = t[:-2]
        for t in self.types:
            name = t
            numpified_zs[name] = (zs[t].squeeze().detach().cpu(
            ).numpy() > 0) if t in zs else np.ones(self.shapes[name])
        return numpified_zs

    def calculate_model_size(self, zs):
        if zs is None: