cf72dcaef367bf5625aa3a83a8bce8bbe823d892,examples/experimental/next-frame.py,VideoFrameSampler,__init__,#VideoFrameSampler#Any#Any#,628

Before Change

class VideoFrameSampler(BaseSampler):
    def __init__(self, gan, samples_per_row=8):
        sess = gan.session
        self.x = gan.session.run(gan.preview)
        print("__________", np.shape(self.x),"---oo")
        frames = np.shape(self.x)[1]//height
        self.frames=frames
        self.x = np.split(self.x, frames, axis=1)
        self.i = 0
        BaseSampler.__init__(self, gan, samples_per_row)

    def _sample(self):
        gan = self.gan
        z_t = gan.uniform_distribution.sample

After Change

        return loss, reg_d1

class VideoFrameSampler(BaseSampler):
    def __init__(self, gan, samples_per_row=8):
        BaseSampler.__init__(self, gan, samples_per_row)
        self.EZ = self.gan.ez
        self.EC = self.gan.ec
        self.G = self.gan.generator
        self.seed()

Instances