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

more debugging

Browse source
This commit is contained in:
jh 2021-03-16 16:57:53 -07:00
commit 6092a77924
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
models/Model_SAEHD/Model.py
View file

@ -497,8 +497,9 @@ Examples: df, liae, df-d, df-ud, liae-ud, ...
print("Shape (gpu_pred_src_src_d): ", gpu_pred_src_src_d.shape)
print("Shape (gpu_pred_src_src_d2): ", gpu_pred_src_src_d2.shape)
for i in gpu_pred_src_src_d2.shape:
print("Shape (gpu_pred_src_src_d2) i: ", i)
print("Flattened Shape (gpu_pred_src_src_d): ", gpu_pred_src_src_d.flatten().shape())
# def get_smooth_noisy_labels_like(label, tensor, smoothing=0.2, noise=0.05):
# new_labels = []