15700dc6bbf845a2f212063982c00642a386d57e,hypergan/samplers/batch_walk_sampler.py,BatchWalkSampler,_sample,#BatchWalkSampler#,24

Before Change

    def _sample(self):
        gan = self.gan
        self.step+=1
        if self.step > self.step_count:
            self.latent1 = self.latent2
            self.latent2 = self.gan.latent.next()
            self.direction = (self.latent2-self.latent1) / self.step_count
            self.step = 0

        latent = self.direction * self.step + self.latent1
        latent = self.hardtanh(latent)
        self.latent2 = latent

After Change

        gan = self.gan

        self.pos = self.direction * self.velocity + self.pos
        mask = torch.gt(self.pos, self.ones)
        mask += torch.lt(self.pos, -self.ones)
        self.direction = self.direction + 2 * self.direction * (-self.ones * mask)

        g = gan.generator.forward(self.pos)

Instances