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

fix range

Browse source
This commit is contained in:
jh 2021-03-16 19:18:22 -07:00
commit 4223ec7986
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

@ -506,7 +506,7 @@ Examples: df, liae, df-d, df-ud, liae-ud, ...
num_labels = self.batch_size num_labels = self.batch_size
for d in tensor.shape[1:]: for d in tensor.shape[1:]:
num_labels *= d num_labels *= d
for _ in num_labels: for _ in range(num_labels):
if np.random.random() < noise: if np.random.random() < noise:
label = 0 if label == 1 else 1 label = 0 if label == 1 else 1
if label == 0: if label == 0: