SAE:fix multiscale decoder

2019-04-20 21:30:40 +04:00
parent 242b850245
commit 7c5f762e78
1 changed files with 2 additions and 2 deletions
@@ -387,9 +387,9 @@ class SAEModel(ModelBase):
    #override
    def onGetPreview(self, sample):
        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_m = sample[1][2][0:4]
+        test_D_m = sample[1][1+self.ms_count][0:4]

        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]) ) ]