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

fix bug

Browse source
This commit is contained in:
jh 2021-05-08 12:27:06 -07:00
commit 3f27107eaf
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[1:])
x = tf.reshape(x, (self.batch_size,) + tensor.shape.as_list()[1:])
return x
smoothing = self.options['gan_smoothing']