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

fix

Browse source
This commit is contained in:
iperov 2019-05-01 19:59:43 +04:00
parent 2a8dd788dc
commit 9577c2d254
1 changed files with 7 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

15
models/Model_SAE/Model.py
View file

@ -389,18 +389,17 @@ class SAEModel(ModelBase):
ar += [ [self.decoderm, 'decoderm.h5'] ]
elif 'df' in self.options['archi']:
ar += [ [self.encoder, 'encoder.h5'],
]
if not self.pretrain or self.iter == 0:
ar += [ [self.decoder_src, 'decoder_src.h5'],
[self.decoder_dst, 'decoder_dst.h5']
ar += [ [self.encoder, 'encoder.h5'],
]
ar += [ [self.decoder_src, 'decoder_src.h5'],
[self.decoder_dst, 'decoder_dst.h5']
]
if self.options['learn_mask']:
if not self.pretrain or self.iter == 0:
ar += [ [self.decoder_srcm, 'decoder_srcm.h5'],
[self.decoder_dstm, 'decoder_dstm.h5'] ]
ar += [ [self.decoder_srcm, 'decoder_srcm.h5'],
[self.decoder_dstm, 'decoder_dstm.h5'] ]
self.save_weights_safe(ar)