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

upd readme

Browse source
This commit is contained in:
iperov 2019-02-05 13:27:18 +04:00
parent 6d95dd4a99
commit 241ae83e0b
2 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
models/Model_SAE/Model.py
View file

@ -318,8 +318,8 @@ class SAEModel(ModelBase):
st = []
for i in range(0, len(test_A)):
ar = S[i], SS[i], D[i], DD[i], SD[i]
if self.options['learn_mask']:
ar += (SDM[i],)
#if self.options['learn_mask']:
# ar += (SDM[i],)
st.append ( np.concatenate ( ar, axis=1) )
return [ ('SAE', np.concatenate (st, axis=0 )), ]