From 7696a30d84133951614cde87f9abe30e631d12c4 Mon Sep 17 00:00:00 2001 From: jh Date: Wed, 17 Mar 2021 08:44:45 -0700 Subject: [PATCH] clean up --- models/Model_SAEHD/Model.py | 23 +++++++---------------- 1 file changed, 7 insertions(+), 16 deletions(-) diff --git a/models/Model_SAEHD/Model.py b/models/Model_SAEHD/Model.py index ff5584a..81368b1 100644 --- a/models/Model_SAEHD/Model.py +++ b/models/Model_SAEHD/Model.py @@ -495,13 +495,7 @@ Examples: df, liae, df-d, df-ud, liae-ud, ... gpu_pred_src_src_d, \ gpu_pred_src_src_d2 = self.D_src(gpu_pred_src_src_masked_opt) - print("gpu_pred_src_src_d.shape: ", gpu_pred_src_src_d.shape) - print("gpu_pred_src_src_d2.shape: ", gpu_pred_src_src_d2.shape) - - # print("tf.reshape(gpu_pred_src_src_d, [-1]).shape: ", tf.reshape(gpu_pred_src_src_d, [-1]).shape) - - - def get_smooth_noisy_labels(label, tensor, smoothing=0.2, noise=0.05): + def get_smooth_noisy_labels(label, tensor, smoothing=0.1, noise=0.05): num_labels = self.batch_size for d in tensor.get_shape().as_list()[1:]: num_labels *= d @@ -515,20 +509,17 @@ Examples: df, liae, df-d, df-ud, liae-ud, ... x = tf.reshape(x, (self.batch_size,) + tensor.shape[1:]) return x - gpu_pred_src_src_d_ones = get_smooth_noisy_labels(1, gpu_pred_src_src_d, smoothing=0.2, noise=0.05) - gpu_pred_src_src_d_zeros = get_smooth_noisy_labels(0, gpu_pred_src_src_d, smoothing=0.2, noise=0.05) + gpu_pred_src_src_d_ones = get_smooth_noisy_labels(1, gpu_pred_src_src_d) + gpu_pred_src_src_d_zeros = get_smooth_noisy_labels(0, gpu_pred_src_src_d) - print("gpu_pred_src_src_d_ones.shape: ", gpu_pred_src_src_d_ones.shape) - print("gpu_pred_src_src_d_zeros.shape: ", gpu_pred_src_src_d_zeros.shape) - - gpu_pred_src_src_d2_ones = get_smooth_noisy_labels(1, gpu_pred_src_src_d2, smoothing=0.2, noise=0.05) - gpu_pred_src_src_d2_zeros = get_smooth_noisy_labels(0, gpu_pred_src_src_d2, smoothing=0.2, noise=0.05) + gpu_pred_src_src_d2_ones = get_smooth_noisy_labels(1, gpu_pred_src_src_d2) + gpu_pred_src_src_d2_zeros = get_smooth_noisy_labels(0, gpu_pred_src_src_d2) gpu_target_src_d, \ gpu_target_src_d2 = self.D_src(gpu_target_src_masked_opt) - gpu_target_src_d_ones = get_smooth_noisy_labels(1, gpu_target_src_d, smoothing=0.2, noise=0.05) - gpu_target_src_d2_ones = get_smooth_noisy_labels(1, gpu_target_src_d2, smoothing=0.2, noise=0.05) + gpu_target_src_d_ones = get_smooth_noisy_labels(1, gpu_target_src_d) + gpu_target_src_d2_ones = get_smooth_noisy_labels(1, gpu_target_src_d2) gpu_D_src_dst_loss = (DLoss(gpu_target_src_d_ones , gpu_target_src_d) + \ DLoss(gpu_pred_src_src_d_zeros , gpu_pred_src_src_d) ) * 0.5 + \