From d4335b5fa5d2342fb92535d1e31d950497594e33 Mon Sep 17 00:00:00 2001 From: Colombo Date: Tue, 18 Feb 2020 19:16:44 +0400 Subject: [PATCH] fix --- models/Model_SAEHD/Model.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/models/Model_SAEHD/Model.py b/models/Model_SAEHD/Model.py index a0eb1f1..98cc400 100644 --- a/models/Model_SAEHD/Model.py +++ b/models/Model_SAEHD/Model.py @@ -644,14 +644,14 @@ class SAEHDModel(ModelBase): self.D_train = D_train if gan_power != 0: - def D_src_dst_train(warped_src, target_src, target_srcm, \ - warped_dst, target_dst, target_dstm): + def D_src_dst_train(warped_src, target_src, target_srcm_all, \ + warped_dst, target_dst, target_dstm_all: nn.tf_sess.run ([src_D_src_dst_loss_gv_op], feed_dict={self.warped_src :warped_src, self.target_src :target_src, - self.target_srcm:target_srcm, + self.target_srcm_all:target_srcm_all, self.warped_dst :warped_dst, self.target_dst :target_dst, - self.target_dstm:target_dstm}) + self.target_dstm_all:target_dstm_all}) self.D_src_dst_train = D_src_dst_train if learn_mask: