edf96731ef964481a23d7c0bbeb0790ce1a7419d,hypergan/distributions/uniform_distribution.py,UniformDistribution,__init__,#UniformDistribution#Any#Any#,12

Before Change

        self.current_channels = config["z"]
        self.current_input_size = config["z"]
        batch_size = gan.batch_size()
        self.z = torch.Tensor(batch_size, self.current_input_size).cuda()
        if self.config.projections is not None:
            self.current_input_size *= len(self.config.projections)
            self.current_channels *= len(self.config.projections)

After Change

        self.current_input_size = config["z"]
        batch_size = gan.batch_size()
        self.shape = [batch_size, self.current_input_size]
        self.const_two = torch.Tensor([2]).cuda()
        self.const_one = torch.Tensor([1]).cuda()
        if self.config.projections is not None:
            self.current_input_size *= len(self.config.projections)
            self.current_channels *= len(self.config.projections)

Instances