self.config.gammas[0],//torch.Tensor([self.config.gammas[0]]).float()[0].cuda(),//self.gan.configurable_param(self.config.gamma or 1.0)
self.config.gammas[1]//torch.Tensor([self.config.gammas[1]]).float()[0].cuda()//self.gan.configurable_param(self.config.gamma or 1.0)
]
self.relu = torch.nn.ReLU()
self.target = [Parameter(x, requires_grad=True) for x in self.gan.discriminator_real_inputs()]
self.x_mod_target = torch.zeros_like(self.target[0])
self.g_mod_target = torch.zeros_like(self.target[0])
After Change
self.config.gammas[1]//torch.Tensor([self.config.gammas[1]]).float()[0].cuda()//self.gan.configurable_param(self.config.gamma or 1.0)
]
self.relu = torch.nn.ReLU()
self.target = [Parameter(x, requires_grad=True) for x in self.gan.discriminator_real_inputs()]
self.x_mod_target = torch.zeros_like(self.target[0])
self.g_mod_target = torch.zeros_like(self.target[0])