From f210ea36003e50c8930cf2b42bfe9e6ac3da1ad7 Mon Sep 17 00:00:00 2001
From: wangna11BD <79366697+wangna11BD@users.noreply.github.com>
Date: Tue, 4 Jan 2022 14:12:56 +0800
Subject: [PATCH] fix lapstyle (#551)

---
 ppgan/models/lapstyle_model.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/ppgan/models/lapstyle_model.py b/ppgan/models/lapstyle_model.py
index bf090ea..b3a644e 100644
--- a/ppgan/models/lapstyle_model.py
+++ b/ppgan/models/lapstyle_model.py
@@ -186,7 +186,7 @@ class LapStyleRevFirstModel(BaseModel):
         self.nets['net_dec'] = build_generator(draftnet_decode)
 
         self.set_requires_grad([self.nets['net_enc']], False)
-        self.set_requires_grad([self.nets['net_enc']], False)
+        self.set_requires_grad([self.nets['net_dec']], False)
 
         # define revision-net params
         self.nets['net_rev'] = build_generator(revnet_generator)
@@ -322,7 +322,7 @@ class LapStyleRevSecondModel(BaseModel):
         self.nets['net_enc'] = build_generator(draftnet_encode)
         self.nets['net_dec'] = build_generator(draftnet_decode)
         self.set_requires_grad([self.nets['net_enc']], False)
-        self.set_requires_grad([self.nets['net_enc']], False)
+        self.set_requires_grad([self.nets['net_dec']], False)
 
         # define the first revnet params
         self.nets['net_rev'] = build_generator(revnet_generator)
-- 
GitLab