github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-08-19 21:13:20 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix

Browse source
This commit is contained in:
jh 2021-05-08 12:37:27 -07:00
commit c2bdf26520
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
models/Model_SAEHD/Model.py
View file

@ -580,7 +580,7 @@ Examples: df, liae, df-d, df-ud, liae-ud, ...
x = tf.cast(x, tf.float32)
x = tf.math.scalar_mul(1-smoothing, x)
# x = x + (smoothing/num_labels)
x = tf.reshape(x, (self.batch_size,) + tensor.shape.as_list()[1:])
x = tf.reshape(x, (self.batch_size,) + tensor.get_shape().as_list()[1:])
return x
smoothing = self.options['gan_smoothing']