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

debug

Browse source
This commit is contained in:
Jeremy Hummel 2019-08-14 11:07:55 -07:00
commit 8f87aff852
1 changed files with 1 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
models/Model_SAE/Model.py
View file

@ -436,10 +436,9 @@ class SAEModel(ModelBase):
else:
if self.options['learn_mask']:
from keras import K
self.AE_convert = K.function([warped_dst], [pred_src_dst[-1][:4], pred_dst_dstm[-1], pred_src_dstm[-1]])
else:
self.AE_convert = K.function([warped_dst], [pred_src_dst[-1]])
self.AE_convert = K.function([warped_dst], [pred_src_dst[-1][:4]])
if self.is_training_mode:
self.src_sample_losses = []