From e9ae398e8e33192733e96c3405e04e3c46ab466a Mon Sep 17 00:00:00 2001 From: jh Date: Wed, 17 Mar 2021 12:03:04 -0700 Subject: [PATCH] test --- models/Model_SAEHD/Model.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/models/Model_SAEHD/Model.py b/models/Model_SAEHD/Model.py index a01836d..be470df 100644 --- a/models/Model_SAEHD/Model.py +++ b/models/Model_SAEHD/Model.py @@ -455,8 +455,9 @@ Examples: df, liae, df-d, df-ud, liae-ud, ... print('abs_loss.shape: ', abs_loss.shape) abs_mean_loss = tf.reduce_mean(300 * abs_loss, axis=[1,2,3]) print('abs_mean_loss.shape: ', abs_mean_loss.shape) + gpu_src_loss += 300*nn.MsSsim(resolution, kernel_size=5)(gpu_target_src*gpu_target_part_mask, gpu_pred_src_src*gpu_target_part_mask, max_val=1.0) - gpu_src_loss += tf.reduce_mean ( 300*nn.MsSsim(resolution, kernel_size=5)(gpu_target_src*gpu_target_part_mask, gpu_pred_src_src*gpu_target_part_mask, max_val=1.0)) + # gpu_src_loss += tf.reduce_mean ( 300*nn.MsSsim(resolution, kernel_size=5)(gpu_target_src*gpu_target_part_mask, gpu_pred_src_src*gpu_target_part_mask, max_val=1.0)) else: gpu_src_loss += tf.reduce_mean ( 300*tf.abs ( gpu_target_src*gpu_target_part_mask - gpu_pred_src_src*gpu_target_part_mask ), axis=[1,2,3])