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

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
models/Model_SAE/Model.py
+2 -2
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]) ) ]