From 4d9609d0ab9cf1058f564e354dfbc29c0ac3108a Mon Sep 17 00:00:00 2001
From: lzzyzlbb <287246233@qq.com>
Date: Mon, 30 Aug 2021 19:26:17 +0800
Subject: [PATCH] Fom mobile (#412)

* add fom lite training

* add fom lite training

* modify according to review

* modify according to review

* fix error of fom trian error
---
 ppgan/models/firstorder_model.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/ppgan/models/firstorder_model.py b/ppgan/models/firstorder_model.py
index 75cb828..baa9d6a 100755
--- a/ppgan/models/firstorder_model.py
+++ b/ppgan/models/firstorder_model.py
@@ -294,14 +294,13 @@ class FirstOrderModelMobile(FirstOrderModel):
             self.nets['kp_detector'] = self.nets[
                 'Gen_Full']._layers.kp_extractor
             self.nets['generator'] = self.nets['Gen_Full']._layers.generator
-            self.kp_detector_ori = self.Gen_Full_ori._layers.kp_extractor
             self.nets['generator'] = self.nets['Gen_Full']._layers.generator
             self.nets['discriminator'] = self.nets['Dis']._layers.discriminator
         else:
             self.nets['kp_detector'] = self.nets['Gen_Full'].kp_extractor
             self.nets['generator'] = self.nets['Gen_Full'].generator
-            self.kp_detector_ori = self.Gen_Full_ori.kp_extractor
             self.nets['discriminator'] = self.nets['Dis'].discriminator
+        self.kp_detector_ori = self.Gen_Full_ori.kp_extractor
 
         from ppgan.utils.download import get_path_from_url
         vox_cpk_weight_url = 'https://paddlegan.bj.bcebos.com/applications/first_order_model/vox-cpk.pdparams'
-- 
GitLab