a178a47269013e94c4756f94a2c9a0c373e04a18,examples/next-frame.py,VideoFrameSampler,seed,#VideoFrameSampler#,556

Before Change

            self.rg = self.G(self.rc, context={"z":self.rz})
        self.g = self.input_cache[self.gan.frames-1]
        self.i=0
        for i in range(self.gan.frames):
            g = self.input_cache[i]
            self.z = self.EZ(g, context={"z":self.z})
            self.c = self.EC(self.z, context={"c":self.c})
            self.g = self.G(self.z)
            self.i = 0

    def refresh_input_cache(self):
        self.input_cache = list(torch.chunk(self.gan.inputs.next(), self.gan.frames, dim=1))
        for i in range(len(self.input_cache)-1):
            self.gan.inputs.next()

After Change

        self.state = self.gan.gen_state()
        print("STATET1", self.state.shape)

        enc = self.gan.encoder(torch.cat(self.input_cache[:-1], dim=1))
        self.state = self.gan.state(enc, context={"past": self.state})
        print("STATET", self.state.shape)
        g = self.gan.decoder(self.state)

Instances