diff --git a/ppgan/models/losses.py b/ppgan/models/losses.py index 3fea20ea7d6fe43ee630e807fdfccd6b4953d2af..6da51e2c3842b0687d4a16ec77274fb4ba865cd8 100644 --- a/ppgan/models/losses.py +++ b/ppgan/models/losses.py @@ -48,13 +48,17 @@ class GANLoss(nn.Layer): """ if target_is_real: if not hasattr(self, 'target_real_tensor'): - self.target_real_tensor = self.target_real_label * paddle.ones( - paddle.shape(prediction), dtype='float32') + self.target_real_tensor = paddle.fill_constant( + shape=paddle.shape(prediction), + value=self.target_real_label, + dtype='float32') target_tensor = self.target_real_tensor else: if not hasattr(self, 'target_fake_tensor'): - self.target_fake_tensor = self.target_fake_label * paddle.ones( - paddle.shape(prediction), dtype='float32') + self.target_fake_tensor = paddle.fill_constant( + shape=paddle.shape(prediction), + value=self.target_fake_label, + dtype='float32') target_tensor = self.target_fake_tensor # target_tensor.stop_gradient = True diff --git a/ppgan/modules/init.py b/ppgan/modules/init.py index 0a9fceef124dd4d7bb153a8f963b08000718a6d4..6b33cc197e89e729c7f11bfc8c69551bf0a2f6d0 100644 --- a/ppgan/modules/init.py +++ b/ppgan/modules/init.py @@ -80,7 +80,7 @@ def calculate_gain(nonlinearity, param=None): @paddle.no_grad() def constant_(x, value): - temp_value = value * paddle.ones(x.shape, x.dtype) + temp_value = paddle.full(shape=x.shape, fill_value=value, dtype=x.dtype) x.set_value(temp_value) return x