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

fix reshape

Browse source
This commit is contained in:
jh 2021-03-16 19:20:24 -07:00
commit a89cf63e9f
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

@ -514,7 +514,7 @@ Examples: df, liae, df-d, df-ud, liae-ud, ...
else:
label = np.random.uniform(1-smoothing/2, 1.0)
labels.append(label)
return tf.reshape(labels, tensor.shape)
return tf.reshape(labels, (self.batch_size,) + tensor.shape[1:])
gpu_pred_src_src_d_ones = get_smooth_noisy_labels(1, gpu_pred_src_src_d, smoothing=0.2, noise=0.05)
gpu_pred_src_src_d_zeros = get_smooth_noisy_labels(0, gpu_pred_src_src_d, smoothing=0.2, noise=0.05)