7066c72279f2fd0f5d0a6b7c3c57e5398dfe7813,examples/colorizer.py,Sampler,sample,#Sampler#Any#Any#,22

Before Change

        sample = sess.run(generator, {x_t: x_v})
        stacks = []
        bs = gan.batch_size()
        width = 5
        print(np.shape(x_v[1]), np.shape(sample[2]), np.shape(layer_filter_v[1]))
        stacks.append([x_v[1], layer_filter_v[1], sample[2], sample[3], sample[4]])
 

After Change

        x_v = np.tile(x_v[0], [gan.batch_size(),1,1,1])
        if layer_filter == None:
            layer_filter = gan.generator.config.layer_filter(gan, gan.generator.config, x_t)
            if(gan.ops.shape(layer_filter)[-1] == 1):
                layer_filter = tf.tile(layer_filter,[1,1,1,3])

        
        layer_filter_v = sess.run(layer_filter, {x_t: x_v})

        samples = [sess.run(generator, {x_t: x_v})[0] for n in range(n_samples)]
        stacks = []

Instances