314f925d0a089d5a44331c95eb2bfefcae1b0bc7,hypergan/samplers/batch_walk_sampler.py,BatchWalkSampler,_sample,#BatchWalkSampler#,29

Before Change

        latent = self.latent2 * self.step / self.step_count + (1.0 - self.step / self.step_count) * self.latent1

        g = gan.generator.forward(latent.cuda().float())
        //    gs.append(g)
        return [
            ("generator", g)

After Change

        if self.step > self.step_count:
            self.latent1 = self.latent2
            direction = self.gan.latent.sample()
            self.direction = direction / torch.norm(direction, p=2, dim=1, keepdim=True).expand_as(direction)
            self.velocity = 10.0
            self.step = 0

        latent = self.direction * self.step / self.step_count * self.velocity + self.latent1

Instances