fb5a8a3e0a73eea528ac398c04ba20a96ef575e1,examples/experimental/next-frame.py,VideoFrameSampler,_sample,#VideoFrameSampler#,156

Before Change

            self.seed()
        self.i += 1
        time.sleep(0.05)
        return [("input", self.gan.inputs.next()), ("generator", self.g)]


class TrainingVideoFrameSampler(BaseSampler):
    def __init__(self, gan, samples_per_row=8):

After Change

            self.i = 0

    def _sample(self):
        self.inp = self.gan.inputs.next()
        samples = [("input", self.inp)]
        self.z = self.EZ(self.g, context={"z":self.z})
        self.c = self.EC(self.z, context={"c":self.c})
        self.g = self.G(self.c, context={"z":self.z})
        if self.i % 100 == 0:
            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
        time.sleep(0.033)
        samples += [("generator", self.g)]
        //time.sleep(0.1)
        return samples


class TrainingVideoFrameSampler(BaseSampler):
    def __init__(self, gan, samples_per_row=8):

Instances