From 62f1d5787140537cd63768e51bd68082a6be312b Mon Sep 17 00:00:00 2001 From: iperov Date: Mon, 14 Jun 2021 17:58:01 +0400 Subject: [PATCH] AMP: fix a merge bug where morph_factor was always = 1.0 --- models/Model_AMP/Model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/Model_AMP/Model.py b/models/Model_AMP/Model.py index ad66365..9347944 100644 --- a/models/Model_AMP/Model.py +++ b/models/Model_AMP/Model.py @@ -586,7 +586,7 @@ class AMPModel(ModelBase): gpu_src_dst_code = tf.concat( ( tf.slice(gpu_dst_inter_src_code, [0,0,0,0], [-1, ae_dims_slice , lowest_dense_res, lowest_dense_res]), tf.slice(gpu_dst_inter_dst_code, [0,ae_dims_slice,0,0], [-1,ae_dims-ae_dims_slice, lowest_dense_res,lowest_dense_res]) ), 1 ) - gpu_pred_src_dst, gpu_pred_src_dstm = self.decoder(gpu_dst_inter_src_code) + gpu_pred_src_dst, gpu_pred_src_dstm = self.decoder(gpu_src_dst_code) _, gpu_pred_dst_dstm = self.decoder(gpu_dst_inter_dst_code) def AE_merge(warped_dst, morph_value):