github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-07 05:22:06 -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

3
models/Model_SAE/Model.py
View file

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