From 6687213fa5a75d71800fe246d0c0e5bcaeb13b8f Mon Sep 17 00:00:00 2001 From: Colombo Date: Fri, 27 Mar 2020 00:15:18 +0400 Subject: [PATCH] SAEHD: fixed the use of pretrained liae model, now it produces less face morphing --- models/Model_SAEHD/Model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/Model_SAEHD/Model.py b/models/Model_SAEHD/Model.py index 2dc443d..549e041 100644 --- a/models/Model_SAEHD/Model.py +++ b/models/Model_SAEHD/Model.py @@ -483,7 +483,7 @@ class SAEHDModel(ModelBase): if model == self.inter: do_init = True elif 'liae' in archi: - if model == self.inter_AB: + if model == self.inter_AB or model == self.inter_B: do_init = True else: do_init = self.is_first_run()