ad0437cbe8b274d5268914718cf036efe5f4b772,hypergan/graph/graph.py,Graph,create_z_encoding,#Graph#,64

Before Change

        return [d_real, d_fake, d_reals, d_fakes]

    def create_z_encoding(self):
        z_base_config = hc.Config(hc.lookup_functions(self.gan.config.z_encoder_base))
        self.gan.graph.z_base = z_base_config.create(z_base_config, self.gan)
        encoders = [self.gan.graph.z_base]
        for i, encoder in enumerate(self.gan.config.z_encoders):
            encoder = hc.Config(hc.lookup_functions(encoder))

After Change

        return [d_real, d_fake, d_reals, d_fakes]

    def create_z_encoding(self):
        self.gan.graph.z = []
        encoders = []
        for i, encoder in enumerate(self.gan.config.encoders):
            encoder = hc.Config(hc.lookup_functions(encoder))
            zs, z_base = encoder.create(encoder, self.gan)
            encoders.append(zs)
            self.gan.graph.z.append(z_base)

        z_encoded = tf.concat(1, encoders)
        self.gan.graph.z_encoded = z_encoded

Instances