david/DeepFaceLab_Linux
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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
models/Model_SAE/Model.py
+7 -8
View File
@@ -389,18 +389,17 @@ 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']:
ar += [ [self.encoder, 'encoder.h5'],
]
if not self.pretrain or self.iter == 0: if not self.pretrain or self.iter == 0:
ar += [ [self.decoder_src, 'decoder_src.h5'], ar += [ [self.encoder, 'encoder.h5'],
[self.decoder_dst, 'decoder_dst.h5']
] ]
ar += [ [self.decoder_src, 'decoder_src.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'] ]
self.save_weights_safe(ar) self.save_weights_safe(ar)