From 0c3559d2dd9c89c6e69f0b6eeb815f54fa342459 Mon Sep 17 00:00:00 2001 From: iperov Date: Wed, 9 Jan 2019 10:46:12 +0400 Subject: [PATCH] fix sae --- models/Model_SAE/Model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/Model_SAE/Model.py b/models/Model_SAE/Model.py index 31c7f88..30d85d5 100644 --- a/models/Model_SAE/Model.py +++ b/models/Model_SAE/Model.py @@ -163,7 +163,7 @@ class SAEModel(ModelBase): style_power = self.options['style_power'] / 100.0 src_loss = K.mean( 100*K.square(tf_dssim(2.0)( target_src_masked, pred_src_src_masked )) ) - #src_loss += tf_style_loss(gaussian_blur_radius=resolution // 8, loss_weight=0.2*style_power)(psd_target_dst_masked, target_dst_masked) + src_loss += tf_style_loss(gaussian_blur_radius=resolution // 8, loss_weight=0.2*style_power)(psd_target_dst_masked, target_dst_masked) src_loss += K.mean( (100*style_power)*K.square(tf_dssim(2.0)( psd_target_dst_anti_masked, target_dst_anti_masked ))) if self.options['archi'] == 'liae':