a178a47269013e94c4756f94a2c9a0c373e04a18,examples/next-frame.py,VideoFrameSampler,_sample,#VideoFrameSampler#,589

Before Change

        if self.i % (8*24) == 0:
            print("RESET")
            self.seed()
        if self.gan.config.random:
            samples += [("rand", self.rg)]
            self.rz = self.EZ(self.rg, context={"z":self.rz})
            self.rc = self.EC(self.rz, context={"c":self.rc})
            self.rg = self.G(self.rc, context={"z":self.rz})
        self.i += 1
        samples += [("generator", self.g)]
        return samples

After Change

        self.next_input()
        samples = []
        inframes = self.current_frames[-args.per_sample_frames+1:]
        enc = self.gan.encoder(torch.cat(inframes, dim=1))
        self.state = self.gan.state(enc, context={"past": self.state})
        g = self.gan.decoder(self.state)
        self.current_frames += [g]
        self.current_frames = self.current_frames[1:]
        self.g = g
        print(self.g.mean())
        if self.i % (8*24) == 0:
            print("RESET")

Instances