This commit is contained in:
2022-06-16 16:45:21 +00:00
parent 81abb9141c
commit e712e456bc
6 changed files with 15 additions and 13 deletions
+1
View File
@@ -102,6 +102,7 @@ class Generator(tf.keras.Model):
w_broadcasted = self.truncation_trick(w_broadcasted, truncation_psi, truncation_cutoff)
image_out = self.synthesis(w_broadcasted)
image_out = image_out + conditions[4]
image_out = image_out * conditions[3][:,tf.newaxis,:,:]
if ret_w_broadcasted: