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

SAE:fix multiscale decoder

Browse source
This commit is contained in:
iperov 2019-04-20 21:30:40 +04:00
parent 242b850245
commit 7c5f762e78
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
models/Model_SAE/Model.py
View file

@ -387,9 +387,9 @@ class SAEModel(ModelBase):
#override #override
def onGetPreview(self, sample): def onGetPreview(self, sample):
test_S = sample[0][1][0:4] #first 4 samples test_S = sample[0][1][0:4] #first 4 samples
test_S_m = sample[0][2][0:4] #first 4 samples test_S_m = sample[0][1+self.ms_count][0:4] #first 4 samples
test_D = sample[1][1][0:4] test_D = sample[1][1][0:4]
test_D_m = sample[1][2][0:4] test_D_m = sample[1][1+self.ms_count][0:4]
if self.options['learn_mask']: if self.options['learn_mask']:
S, D, SS, DD, DDM, SD, SDM = [ np.clip(x, 0.0, 1.0) for x in ([test_S,test_D] + self.AE_view ([test_S, test_D]) ) ] S, D, SS, DD, DDM, SD, SDM = [ np.clip(x, 0.0, 1.0) for x in ([test_S,test_D] + self.AE_view ([test_S, test_D]) ) ]